no-cuda error
oxm opened this issue · 2 comments
@ShivamDuggal4
args.cuda = True,submission_kitti.py runs correctly,when I set --no-cuda,args.cuda=False,such error appeared:
Traceback (most recent call last):
File "submission_kitti.py", line 83, in
model.load_state_dict(state_dict['state_dict'], strict=True)
File "/home/xxm/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 845, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DeepPruner:
Hi @oxm , Just take model = nn.DataParallel(model) out of the if args.cuda: block in the line
DeepPruner/deeppruner/submission_kitti.py
Line 73 in 2b5a0e9
I will edit it in some of the later commits.
@ShivamDuggal4 hi,according to your opinion, the problem was sovled.