slothkong / DNN-Pruning

A pyCaffe implementaion of the 2017 ICLR's "Pruning Filters for Efficient ConvNets" publication
43 stars 0 forks source link

Check failed: axis_index < num_axes() (1 vs. 1) axis 1 out of range for 1-D Blob with shape 1 (1) #3

Closed 408550969 closed 6 years ago

408550969 commented 6 years ago

The caffe report this error when I use this command: python3 filter_pruning_demo.py -model ./temp/model.prototxt -weights ./temp/1.caffemodel or python3 filter_pruning_demo.py -model ./temp/model.prototxt How to solve this problem?Thank you!

--- saveModel --- in pt (new_pt): ./temp/renamed_model.prototxt in model: ./temp/1.caffemodel in WPQ: dict_keys([('conv4_1_P', 1), ('conv5_2_P', 1), 'conv3_1_scale_P', ('conv4_2_P', 1), 'conv3_2_bn_P', 'conv5_1_bn_P', 'conv5_1_scale_P', 'fc_P', 'conv4_3_bn_P', 'conv3_1_bn_P', ('conv5_1_P', 0), ('conv3_1_P', 1), 'conv4_2_scale_P', ('conv4_1_P', 0), 'conv5_3_scale_P', 'conv2_2_scale_P', ('conv4_2_P', 0), 'conv2_2_bn_P', 'conv2_1_scale_P', ('conv1_1_P', 0), ('conv3_1_P', 0), ('conv3_3_P', 1), 'conv4_1_scale_P', 'conv2_1_bn_P', ('conv3_2_P', 0), ('conv1_2_P', 0), ('conv2_2_P', 1), 'conv4_2_bn_P', ('conv2_2_P', 0), 'conv3_3_bn_P', 'conv1_1_bn_P', ('conv2_1_P', 0), ('conv5_3_P', 0), ('conv1_1_P', 1), ('conv4_3_P', 0), 'conv4_3_scale_P', ('conv3_3_P', 0), 'conv1_1_scale_P', ('conv1_2_P', 1), 'conv3_3_scale_P', 'conv1_2_scale_P', ('conv5_1_P', 1), 'conv3_2_scale_P', 'conv4_1_bn_P', ('conv5_2_P', 0), ('conv2_1_P', 1), 'conv5_2_scale_P', ('conv3_2_P', 1), ('conv4_3_P', 1), 'conv5_2_bn_P', 'conv5_3_bn_P', ('conv5_3_P', 1), 'conv1_2_bn_P']) WARNING: Logging before InitGoogleLogging() is written to STDERR F0419 17:33:12.103587 1048 blob.hpp:122] Check failed: axis_index < num_axes() (1 vs. 1) axis 1 out of range for 1-D Blob with shape 1 (1) Check failure stack trace: