Merge pull request #723 from AshishSardana/asardana/efficientnet

narendasan · web-flow · commit c186d81d256b · 2021-11-15T16:54:12.000-05:00
Example notebook for EfficientNet
diff --git a/docsrc/index.rst b/docsrc/index.rst
@@ -53,6 +53,7 @@ Getting Started
    _notebooks/Resnet50-example
    _notebooks/ssd-object-detection-demo
    _notebooks/vgg-qat
+   _notebooks/EfficientNet-example 
 
 
 Python API Documenation
diff --git a/notebooks/EfficientNet-example.ipynb b/notebooks/EfficientNet-example.ipynb
diff --git a/notebooks/README.md b/notebooks/README.md
@@ -48,6 +48,7 @@ Within the container, the notebooks themselves are located at `/Torch-TensorRT/n
 - [ssd-object-detection-demo.ipynb](ssd-object-detection-demo.ipynb): demo for compiling a pretrained SSD model using Torch-TensorRT.
 - [Resnet50-example.ipynb](Resnet50-example.ipynb): demo on the ResNet-50 network.
 - [vgg-qat.ipynb](vgg-qat.ipynb): Quantization Aware Trained models in INT8 using Torch-TensorRT
+- [EfficientNet-example.ipynb](EfficientNet-example.ipynb): Simple use of 3rd party PyTorch model library
 
 ```python
 
