pytorch计算模型参数量报错:size mismatch for module.conv1.weight: copying a param with shape torch.Size([16, 3
错误:RuntimeError:Error(s)inloadingstate_dictforDataParallel:sizemismatchformodule.conv1.weight:copyingaparamwithshapetorch.Size([16,3,3,3])fromcheckpoint,theshapeincurrentmodelistorch.Size([128,128,3,3