Merge pull request #165 from shashank3959/cherry-branch

narendasan · web-flow · commit 489516c7accb · 2020-08-26T14:10:19.000-06:00
Add SSD - TRTorch demo
diff --git a/docsrc/index.rst b/docsrc/index.rst
@@ -48,6 +48,7 @@ Notebooks
    :hidden:
 
    _notebooks/lenet-getting-started
+   _notebooks/ssd-object-detection-demo
 
 
 Python API Documenation
diff --git a/notebooks/README.md b/notebooks/README.md
@@ -42,4 +42,5 @@ Within the container, the notebooks themselves are located at `/workspace/TRTorc
 
 ## 2. Notebook list
 
-- [LeNet-example.ipynb](LeNet-example.ipynb): simple example on a LeNet network.
+- [lenet-getting-started.ipynb](lenet-getting-started.ipynb): simple example on a LeNet network.
+- [ssd-object-detection-demo.ipynb](ssd-object-detection-demo.ipynb): demo for compiling a pretrained SSD model using TRTorch.
diff --git a/notebooks/ssd-object-detection-demo.ipynb b/notebooks/ssd-object-detection-demo.ipynb
