Converting a PyTorch ONNX model to TensorRT engine – Jetson Orin Nano
I’m trying to convert a ViT-B/32 Vision Transformer model from the UNICOM repository on a Jetson Orin Nano. The model’s Vision Transformer class and source code is here.
I’m trying to convert a ViT-B/32 Vision Transformer model from the UNICOM repository on a Jetson Orin Nano. The model’s Vision Transformer class and source code is here.