will-qzq
will-qzq
> `from data.config import coco, voc` > > 将其添加到python文件的顶部。 添加后仍然报错，请问该如何解决
> 你的版本是否正确啊 是1.5的做了相应的修改结果报错
> 具体的 File "/home/ssd-pytorch-master/predict.py", line 12, in ssd = SSD() File "/home/ssd-pytorch-master/ssd.py", line 54, in __init__ self.generate() File "/home/ssd-pytorch-master/ssd.py", line 81, in generate model.load_state_dict(torch.load(self.model_path, map_location=device)) File "/root/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 847, in...
> 具体的 RuntimeError: Error(s) in loading state_dict for SSD: size mismatch for conf.0.weight: copying a param with shape torch.Size([16, 512, 3, 3]) from checkpoint, the shape in current model is...
> num_classes还是classes_path没修改好噢 问题解决，谢谢