【Pytorch实用教程】深入了解 torchvision.models.resnet18 新旧版本的区别
深入了解 torchvision.models.resnet18 新旧版本的区别
在深度学习模型开发中，PyTorch 和 torchvision 一直是我们不可或缺的工具。近期，torchvision 对其模型加载 API 进行了更新，将旧版的 pretrained 参数替换为新的 weights 参数。本文将介绍这一变化的背景、具体区别，以及如何在代码中适配新版本的 torchvision.models.resnet18。
一、背景
随着深度学习模型的不断发展，torchvision 的模型库中预训练权重的种类逐渐增多。为了更灵活、更明确地管理这些权重，开发者决定将 pretrained 参数替换为更具描述性的 weights 参数。
-
旧版：
pretrained参数- 仅支持布尔值：
True表示加载默认的预训练权重，False表示不加载预训练权重。 - 模糊：没有明确说明具体加载了哪个版本的权重。
- 仅支持布尔值：
-
新版：
weights参数</