- 博客(1)
- 收藏
- 关注
原创 pytorch 调用模型
利用pytorch调用预训练好的模型到GPU上 modeldata = torch.load(‘resnet-34-kinetics.pth’) 利用pytorch调用预训练好的模型到CPU上 modeldata = torch.load(‘resnet-34-kinetics.pth’,map_location=lambda storage, loc: storage) 返回值modeldata...
2018-11-19 15:14:32 4179
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人