美文网首页
pytorch:云服务中加载VGG19等模型

pytorch:云服务中加载VGG19等模型

作者: 不太聪明的亚子 | 来源:发表于2020-08-12 10:43 被阅读0次

    之前在本地使用VGG19预训练模型,可以手动下载.pth文件到本地.chche中,之后再使用torchvision.models.vgg19(pretrained = True)

    就会自动从cache中读取。

    现在要在GPU云服务中使用到VGG19模型,如果还是使用这条代码,就会非常缓慢的下载,等下次使用时,还得缓慢下载,所以,需要上传.pth文件到项目目录中,然后设置读取.pth文件,加载模型。

    vgg_19 = torchvision.models.vgg19(pretrained = False)

    vgg_19.load_state_dict(torch.load(pthfile))

    这两句代码就解决了,pthfile就是.pth文件的路径。

    相关文章

      网友评论

          本文标题:pytorch:云服务中加载VGG19等模型

          本文链接:https://www.haomeiwen.com/subject/lpjpdktx.html