BVLC / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
34.03k stars 18.7k forks source link

nn_quantizer.py in the ML-CIFAR10 examples got error KeyError: 'accuracy' #6847

Open rafakath2020 opened 4 years ago

rafakath2020 commented 4 years ago

I am executing ML-examples ->cmsisnn-cifar10. While converting trained Caffe model into cmsis-nn by below command

python nn_quantizer.py --model models/cifar10_m7_train_test.prototxt --weights models/cifar10_m7_iter_300000.caffemodel.h5 --save models/cifar10_m7.pkl

getting an error message.

Traceback (most recent call last): File "nn_quantizer.py", line 614, in my_model.get_graph_connectivity() File "nn_quantizer.py", line 231, in get_graph_connectivity num_branch = len(self.bottom_blob[current_layer]) KeyError: 'accuracy'

Kindly provide any reference for solving the above error.