Retrain pruned model

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Hi,
How to retrain a pruned model in classification?
i.e Which spec configuration need to change?

2

Hi sathiez,
Please refer to the notebook inside the docker. The re-trained spec is also available at each notebook.

3

Where is the Jupyter notebook file you’re referring to? Can you refer me to any documentation about this and/or other notebooks in the Docker container? Thanks in advance…

4

See details in Integrating TAO Models into DeepStream — TAO Toolkit 3.22.05 documentation for triggering docker and Jupyter notebooks.
There are four notebooks: classification, detectnet_v2, faster_rcnn and SSD.
The clssification network’s spec locates at /workspace/examples/classification/specs

5

Thanks, Morgan, very helpful!