pytorch实现从本地加载.pth格式模型
2015年湖南公务员考试职位表可以从官⽹加载预训练好的模型:
东钱湖中学dels as models
model = models.vgg16(pretrained = True)
print(model)
但是经常会出现因为下载速度太慢⽽出现ptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于连接⽅在⼀段时间后没有正确答复或连接的主机没有反应,连接尝试失败。', None, 10060, None))这种错误,因此需要我们⼿动去下载 .pth ⽂件(百度云也很慢,如果你是SVIP,当我没说;迅雷的速度也还可以),然后从本地加载。
吉林省会计网报名入口从本地加载只需要把上⾯的代码换成如下:
dels as models
如何报考成人大专学历model = models.vgg16(pretrained=False)
pre=torch.load(r'.\kaggle_dog_vs_cat\pretrain\vgg16-397923af.pth')
公务员考试科目分类model.load_state_dict(pre)
如果你模型不是⽤的vgg16,⽽是⽤的vgg11或者vgg13,只需要修改语句 model = models.vgg16(pretrained=False) 为对应模型的函数即可。
以上这篇pytorch实现从本地加载 .pth 格式模型就是⼩编分享给⼤家的全部内容了,希望能给⼤家⼀个参考,也希望⼤家多多⽀持。
本地58同城