I’ve been following the guide on training and deploying object detection models with the tlt jupyter notebook examples. I’m following the yolo_v4 resnet18 example.
The training and export work fine, but when I try converting the model to TensorRT on the Xavier AGX it fails:
$ ./tlt-converter -k KEY -d 3,640,640 -o BatchedNMS -c cal.bin -t int8 -i nchw yolov4_resnet18_epoch_080.etlt [ERROR] UffParser: Validator error: FirstDimTile_2: Unsupported operation _BatchTilePlugin_TRT [ERROR] Failed to parse the model, please check the encoding key to make sure it's correct [ERROR] Network must have at least one output [ERROR] Network validation failed. [ERROR] Unable to create engine Segmentation fault (core dumped)
The Xavier board is updated to jetson 4.5 using the SDK manager and I’m using the appropriate conversion script for jetson 4.5.
The conversion works if using the PC that did the training.
Thank you in advance.