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

16 views
Skip to first unread message

Mohammad Rafakath

unread,
Oct 11, 2019, 6:16:56 AM10/11/19
to Caffe Users
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 <module>
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 refernce for solving above error.

Reply all
Reply to author
Forward
0 new messages