TRT Yolo PyTorch Interface