【Pytorch实用教程】深入了解 torchvision.models.resnet18 新旧版本的区别

深入了解 torchvision.models.resnet18 新旧版本的区别

在深度学习模型开发中，PyTorch 和 torchvision 一直是我们不可或缺的工具。近期，torchvision 对其模型加载 API 进行了更新，将旧版的 pretrained 参数替换为新的 weights 参数。本文将介绍这一变化的背景、具体区别，以及如何在代码中适配新版本的 torchvision.models.resnet18。

一、背景

随着深度学习模型的不断发展，torchvision 的模型库中预训练权重的种类逐渐增多。为了更灵活、更明确地管理这些权重，开发者决定将 pretrained 参数替换为更具描述性的 weights 参数。

• 旧版：pretrained 参数

  • 仅支持布尔值：True 表示加载默认的预训练权重，False 表示不加载预训练权重。
  • 模糊：没有明确说明具体加载了哪个版本的权重。

• 新版：weights 参数</