vscode运行yolov3的[predict.py]时出现RuntimeError: Error(s) in loading state_dict for YoloBody:size mismatch for last_layer……的问题解决
在vscode的pytorch环境下运行yolov3的[predict.py]时出现如下问题
RuntimeError: Error(s) in loading state_dict for YoloBody:
size mismatch for last_layer0.6.weight: copying a param with shape torch.Size(【255, 1024, 1, 1】) from checkpoint, the shap