VGG参数由pretrained修改为weights
1.首先修改参数
import torchvision vgg16_false = torchvision.models.vgg16(weights=None) vgg16_true = torchvision.models.vgg16(weights=torchvision.models.VGG16_Weights.DEFAULT)
2.然后将之前下载的vgg文件删除(如果有的话)
3.然后执行程序,一定要关闭外网vpn
第三步的位置可以通过配置文件改变:
打开 D:\Anaconda3_Environments\envs\PyTorch\Lib\site-packages\torch
中的 hub.py
文件,搜索 load_state_dict_from_url
,然后修改 model_dir
即可: