Platform for converting to .wts file

wang-xinyu / tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API

MIT License

7.04k stars 1.78k forks source link

Platform for converting to .wts file #1548

Closed eegmnn closed 4 months ago

eegmnn commented 5 months ago

Env

Xavier AGX
Ubuntu20.04
Cuda version 11.4
TensorRT version 8.4.1

About the repo

Main branch
YoloV9 model

Does it matter what platform I am converting my model.pt file into model.wts file? For example, can I convert pt file into wts file on google colab, and then convert the wts file into engine file on xavier agx?

wang-xinyu commented 4 months ago

.wts is plain text file, it's not related to the hardware. You can convert it on any device.