because of size of model, i want to run the mdel in deplabv3_resnet50. but there are some error. how can i do that? do i change the structure of code?
Traceback (most recent call last):
File "/deeplabv3/main.py", line 188, in
main()
File "/deeplabv3/main.py", line 70, in main
model = getattr(deeplab, 'resnet50')(
File "/deeplabv3/deeplab.py", line 212, in resnet50
model.load_state_dict(model_zoo.load_url(model_urls['resnet50']))
File "/anaconda3/envs/vision/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2041, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for ResNet:
Missing key(s) in state_dict: "aspp.aspp1.weight", "aspp.aspp2.weight", "aspp.aspp3.weight", "aspp.aspp4.weight", "aspp.aspp5.weight", "aspp.aspp1_bn.weight", "aspp.aspp1_bn.bias", "aspp.aspp1_bn.running_mean", "aspp.aspp1_bn.running_var", "aspp.aspp2_bn.weight", "aspp.aspp2_bn.bias", "aspp.aspp2_bn.running_mean", "aspp.aspp2_bn.running_var", "aspp.aspp3_bn.weight", "aspp.aspp3_bn.bias", "aspp.aspp3_bn.running_mean", "aspp.aspp3_bn.running_var", "aspp.aspp4_bn.weight", "aspp.aspp4_bn.bias", "aspp.aspp4_bn.running_mean", "aspp.aspp4_bn.running_var", "aspp.aspp5_bn.weight", "aspp.aspp5_bn.bias", "aspp.aspp5_bn.running_mean", "aspp.aspp5_bn.running_var", "aspp.conv2.weight", "aspp.bn2.weight", "aspp.bn2.bias", "aspp.bn2.running_mean", "aspp.bn2.running_var", "aspp.conv3.weight", "aspp.conv3.bias".
Unexpected key(s) in state_dict: "fc.weight", "fc.bias".
because of size of model, i want to run the mdel in deplabv3_resnet50. but there are some error. how can i do that? do i change the structure of code?
Traceback (most recent call last):
File "/deeplabv3/main.py", line 188, in
main()
File "/deeplabv3/main.py", line 70, in main
model = getattr(deeplab, 'resnet50')(
File "/deeplabv3/deeplab.py", line 212, in resnet50
model.load_state_dict(model_zoo.load_url(model_urls['resnet50']))
File "/anaconda3/envs/vision/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2041, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for ResNet:
Missing key(s) in state_dict: "aspp.aspp1.weight", "aspp.aspp2.weight", "aspp.aspp3.weight", "aspp.aspp4.weight", "aspp.aspp5.weight", "aspp.aspp1_bn.weight", "aspp.aspp1_bn.bias", "aspp.aspp1_bn.running_mean", "aspp.aspp1_bn.running_var", "aspp.aspp2_bn.weight", "aspp.aspp2_bn.bias", "aspp.aspp2_bn.running_mean", "aspp.aspp2_bn.running_var", "aspp.aspp3_bn.weight", "aspp.aspp3_bn.bias", "aspp.aspp3_bn.running_mean", "aspp.aspp3_bn.running_var", "aspp.aspp4_bn.weight", "aspp.aspp4_bn.bias", "aspp.aspp4_bn.running_mean", "aspp.aspp4_bn.running_var", "aspp.aspp5_bn.weight", "aspp.aspp5_bn.bias", "aspp.aspp5_bn.running_mean", "aspp.aspp5_bn.running_var", "aspp.conv2.weight", "aspp.bn2.weight", "aspp.bn2.bias", "aspp.bn2.running_mean", "aspp.bn2.running_var", "aspp.conv3.weight", "aspp.conv3.bias".
Unexpected key(s) in state_dict: "fc.weight", "fc.bias".