Traceback (most recent call last):


 "/root/anaconda3/bin/mmtomodel", line 10, in <module>

sys.exit(_main())


 "/root/anaconda3/lib/python3.7/site-packages/mmdnn/conversion/_script/dump_code.py", line 79, in _main

ret = dump_code(args.framework, args.inputNetwork, args.inputWeight, args.outputModel, args.dump_tag)


 "/root/anaconda3/lib/python3.7/site-packages/mmdnn/conversion/_script/dump_code.py", line 32, in dump_code

save_model(MainModel, network_filepath, weight_filepath, dump_filepath)


 "/root/anaconda3/lib/python3.7/site-packages/mmdnn/conversion/pytorch/saver.py", line 5, in save_model

model = MainModel.KitModel(weight_filepath)


 "Pytorch_VGGFACE_IR.py", line 27, in __init__

self.conv1_1 = self.__conv(2, name='conv1_1', in_channels=3, out_channels=64, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=True)


 "Pytorch_VGGFACE_IR.py", line 129, in __conv

layer.state_dict()['bias'].copy_(torch.from_numpy(__weights_dict[name]['bias']))
RuntimeError: output with shape [64] doesn't match the broadcast shape [1, 1, 1, 64]

Traceback (most recent call last):


 "pre.py", line 65, in <module>

net.load_state_dict(torch.load(weightPath))


 "/home/Software/anaconda3/envs/torch-py3.7/lib/python3.7/site-packages/torch/nn/modules/module.py", line 719, in load_state_dict

self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DataParallel:
