1.直接使用默认程序里的下载方式,往往比较慢;
2.通过修改源代码,使得模型加载已经下载好的参数,修改地方如下:
通过查找自己代码里所调用网络的类,使用pycharm自带的函数查找功能(ctrl+鼠标左键),查看此网络的加载方法,修改model.load_state_dict()函数。
例如:已经下载好的resnet50的参数文件:放在model_urls里面,这样就可以提前下载直接使用。
model_urls = {
'resnet50': '/home/huihua/NewDisk1/pretrain_parameter/resnet50-19c8e357.pth',
}