Optimize Torch linear model with custom arguments for conversion#
In this example, we show how to use custom_args to pass additional arguments for model export and conversion. We recommend running this example in NVIDIA NGC PyTorch container. To run the example, simply run the optimize.py script:
./optimize.py