我已经培训过Torchvision AlexNet with this script。
现在,我加载训练后的权重并将其用于预测val集,我得到的准确度很低。
这是我加载重量的方式:
alexnet = models.alexnet()
model_path = "model_best.pth.tar"
state_dict = torch.load(model_path,map_location='cpu')
alexnet.load_state_dict(state_dict,strict=False)
alexnet.eval()
我知道我的其余代码是正确的,因为它可以与pytorch上的预训练alexnet一起使用。
有什么建议吗?