Optimize and Profile Torch linear model#

In this example, we show how to optimize torch linear model and profile the generated models on a separate dataloader.

We recommend running this example in NVIDIA NGC PyTorch container.

To run the example, simply run the optimize.py script:

./optimize.py