some doubt about convert to TensorRT engine?

ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms

GNU General Public License v2.0

718 stars 208 forks source link

some doubt about convert to TensorRT engine? #260

Closed jcyhcs closed 2 years ago

jcyhcs commented 2 years ago

hi,professor: i have a weights file trained in darknet, with cfg resolution 608x608, when i use tkDNN to infer, think about speed, i may convert 416x416 tensorRT engine, so , can i just use the trained weights in 608*608? is there any modify should i do? please help!

mive93 commented 2 years ago

Hi @jcyhcs,

Yes you can. When you export the weights in tkDNN format, you should change the width and height of the cfg file to 416x416.

mive93 commented 2 years ago

Closing for inactivity