UserWarning: The parameter ‘pretrained‘ is deprecated since 0.13 and may be removed in the future解决

问题描述:

在和土堆学习pytorch时,在网络模型的保存与读取学习保存模型 方式时使用到以下代码:

import torch
import torchvision.models
 
vgg16 = torchvision.models.vgg16(pretrained=False)  # 网络中模型的参数是没有经过训练的、初始化的参数

 相信最近 (2022年7月) 安装或者更新了 PyTorch 和 torchvision 的同志们可能跑代码时遇到了下面两个UserWarning报错:

UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead.

UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=None`.
  warnings.warn(msg)

解决方法:

我的解决方法:删除 pretrained=False

结果:运行成功。

出现问题原因:

这是因为 torchvision 0.13对预训练模型加载方式作出了重大更新造成的。从 torchvision 0.13开始,torchvision提供一个全新的多权重支持API (Multi-weight support API) ,支持将不同版本的权重参数文件加载到模型中。

你可能感兴趣的:(【我是土堆,-,Pytorch教程】,知识点,学习总结笔记,pytorch,人工智能,python,深度学习)