Getting error while running code_gen.py with MNIST

Discussion related to "under the hood" OpenMV topics.
rafakath
Posts: 5
Joined: Mon Oct 14, 2019 7:51 am

Getting error while running code_gen.py with MNIST

Postby rafakath » Tue Nov 12, 2019 7:18 am

hi,
While converting .pkl file into header file by running script code_gen.py with MNIST dataset I am getting below error.

Traceback (most recent call last):
File "code_gen.py", line 362, in <module>
generate_parameters(my_model, cmd_args.out_dir+'/parameter.h')
File "code_gen.py", line 94, in generate_parameters
f.write("#define "+layer.upper()+"_OUT_DIM "+str(caffe_model.layer_shape[layer][2])+"\n\n")
IndexError: tuple index out of range

I am new to Caffe, kindly give any suggestion.

Return to “Technical Discussion”

Who is online

Users browsing this forum: No registered users and 1 guest