使用PyTorch的AlexNet在训练后显示出较低的准确性

时间:2019-04-23 23:51:22

标签: python

我已经培训过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一起使用。

有什么建议吗?

0 个答案:

没有答案
相关问题