- 博客(1)
- 收藏
- 关注
原创 PyTorch加载预训练模型的问题
解决方法:!wget http://download.pytorch.org/models/resnet18-5c106cde.pth然后:import torchimport torchvision.models as modelsresnet18 = models.resnet18()resnet18.load_state_dict(torch.load('resnet18-5c106cde....
2018-06-02 13:06:39 4924 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人