Traceback (most recent call last):
File "/MedMamba/ConfusionMatrix/main.py", line 104, in
net.load_state_dict(torch.load(model_weight_path, map_location=device))
File "/home/user/tools/miniforge/envs/deeplearning/lib/python3.12/site-packages/torch/nn/modules/module.py", line 2215, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for MobileNetV2:
在python train.py之后,运行main文件,出现上述问题,请问该如何调整,期待您的回答!
Traceback (most recent call last): File "/MedMamba/ConfusionMatrix/main.py", line 104, in
net.load_state_dict(torch.load(model_weight_path, map_location=device))
File "/home/user/tools/miniforge/envs/deeplearning/lib/python3.12/site-packages/torch/nn/modules/module.py", line 2215, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for MobileNetV2:
在python train.py之后,运行main文件,出现上述问题,请问该如何调整,期待您的回答!