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