Getting error while running code_gen.py with MNIST

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
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.