airpack_scripts.pytorch.make_plan_file

Module Contents

airpack_scripts.pytorch.make_plan_file.convert(onnx_file)

Converts an onnx file to an optimized plan file using TensorRT. This is a wrapper for airpack.deploy.trt.onnx2plan() that is setup with default values for the airpack.pytorch.model.Network.

See also

If non-default parameters are needed for your customization, it is better practice to call airpack.deploy.trt.onnx2plan() instead of this demonstration wrapper for the airpack.pytorch.model.Network.

Parameters

onnx_file (Union[str, os.PathLike]) – Trained neural network saved as an onnx file

Returns

Name of saved .plan file

Return type

os.PathLike