- 博客(1)
- 收藏
- 关注
原创 pytorch 加载入已下载的预训练模型及参数
#下载预训练模型 cnn = models.vgg19(pretrained=True).features.to(device).eval() 等价于: pretrained=False cnn= models.vgg19(pretrained=False).features.to(device).eval() if pretrained: state_dict = to...
2020-02-14 21:40:48 512
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人