Optimize and profile torch custom pipeline#
In this example, we show how to build a custom pipeline and optimize it with Model Navigator.
We recommend running this example in NVIDIA NGC PyTorch container.
The Python script optimize.py wraps the Python model using Inplace Optimize and then runs profiling.
To run the optimization and profiling, run the script:
./optimize.py