DIMENSIONS WHILE CREATING RT FILE

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 209 forks source link

DIMENSIONS WHILE CREATING RT FILE #205

Closed bobbilichandu closed 2 years ago

bobbilichandu commented 3 years ago

How to change the dimensions while creating .rt file? One way I am using currently is by changing the height and width parameters in cfg while exporting layers from weights and cfg file, but I am not sure if it's correct way. Is there any environment variable like batchsize, FP16 mode ? Thanks in Advance!

bobbilichandu commented 3 years ago

@mive93 could you help me with this?

Sudhakar17 commented 3 years ago

@chandu1263 and @mive93 : Did you find the solution?

mive93 commented 3 years ago

Hi @chandu1263 and @Sudhakar17, if you want to change the input dimension of the network you have to export again the weights form darknet. Follow this guide: https://github.com/ceccocats/tkDNN#1export-weights-from-darknet. Just be sure to put the wanted dimension in the cfg you use for the export.

mive93 commented 2 years ago

Closing for inactivity. Feel free to reopen.