Open OnlyForWW opened 9 months ago
Have you solved this issue? I'm facing s similar problem.
Traceback (most recent call last):
File "./tools/train.py", line 162, in
config file with : load_from = '/home/mchmu/datasets/DP/RSCD-dataset-1million/resnet18_8xb32_in1k_20210831-fbbb1da6.pth'
It might be that the "message_hub" wasn't created during the model training. I commented out this line:
and now it runs fine. However, I'm not sure if there will be any other issues.
分支
main 分支 (mmpretrain 版本)
描述该错误
Hello,I have some questions about model loading weights, I have downloaded the pre-trained model of MMPretrain locally in advance, but when I use torch.load_state_dict() it gives an error, I would like to know how to solve this problem or do I have to load the pre-trained model the way it is documented.
Here's my code.
When I run it, I get the following error.
环境信息
其他信息
No response