Traceback (most recent call last):
File "layer_channel_prune.py", line 31, in
model.load_state_dict(torch.load(opt.weights, map_location=device)['model'])
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 839, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for Darknet:
Traceback (most recent call last): File "layer_channel_prune.py", line 31, in
model.load_state_dict(torch.load(opt.weights, map_location=device)['model'])
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 839, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for Darknet: