我无法下载和使用我之前从在线存储库中保存的模型。代码如下: model = Model().double() # Model is defined in another class
state_dict = torch.hub.load_state_dict_from_url(r'https://filebin.net/j2977ux7kts41aft/checkpoint_best.pt?t=wjbujfoo')
model.load_state_dict(state_dict)
model.eval() 这给出了以下错误: Traceback (most recent
我对CNN非常陌生,在学习上遇到了很多困难。
我试着用ResNet-101提取CNN的特征图,我希望得到一个2048,14*14的形状。为了得到一个特征图,我删除了ResNet-101模型的最后一层,并调整了自适应平均池。所以我得到了输出的torch.Size([1, 2048, 1, 1])形状。
但我想要的是torch.Size([1, 2048, 14, 14])而不是torch.Size([1, 2048, 1, 1])。
有人能帮我得到结果吗?谢谢。
#load resnet101 model and remove the last layer
model = torch.hub.l