这个警告信息是 torchvision 库中的一个警告,提醒你在使用'ResNet101'模型时传递给'weights'参数的值已被弃用,即在0.13版本中被弃用,并在0.15版本中将被移除。警告说明当前行为相当于传递'weights=ResNet101_Weights.IMAGENET1K_V1',你可以使用'weights=ResNet101_Weights.DEFAULT'来获取最新的权重。

可能的解决方法是更新 torchvision 库到最新版本,以避免这个警告。你可以使用以下命令通过 pip 来更新 torchvision 库:

pip install --upgrade torchvision

如果你使用的是 conda 环境,可以使用以下命令来更新 torchvision 库:

conda update torchvision

这将会安装最新版本的 torchvision 库,其中包含了对'weights'参数的更新和修复。

请注意,更新库可能会引入其他新的改变和依赖关系,请确保你的代码在更新之后仍然正常工作,并根据需要进行适当的调整。

解决 PyTorch torchvision 中 ResNet101 模型的'weights'参数警告

原文地址: http://www.cveoy.top/t/topic/bPtx 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录