1.直接使用默认程序里的下载方式,往往比较慢;

2.通过修改源代码,使得模型加载已经下载好的参数,修改地方如下:

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

通过查找自己代码里所调用网络的类,使用pycharm自带的函数查找功能(ctrl+鼠标左键),查看此网络的加载方法,修改model.load_state_dict()函数。

例如:vgg16.load_state_dict(torch.load('vgg16-397923af.pth'))

 pytorch加载预训练模型参数的方式 随笔

 

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄