ceccocats/tkDNN

DIMENSIONS WHILE CREATING RT FILE

Closed this issue · 4 comments

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!

@mive93 could you help me with this?

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

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.

Closing for inactivity. Feel free to reopen.