培训参考¶
除了许多模型、数据集和图像转换之外,Torchvision 还提供了培训参考脚本。这些是我们用于培训 模型 的脚本,这些模型随后可提供预训练权重。
这些脚本不属于核心包,而是可以在 GitHub 上找到。我们目前提供分类、检测、分割、相似性学习和视频分类的参考脚本。
尽管这些脚本大体上是稳定的,但它们不提供向后兼容性保证。
总的来说,这些脚本依赖于最新的(尚未发布的)pytorch 版本或最新的 torchvision 版本。这意味着,要使用它们,您可能需要按照 官方说明 安装最新的 pytorch 和 torchvision 版本。
如果您需要依赖旧的稳定版 pytorch 或 torchvision,例如 torchvision 0.10,那么使用 GitHub 上相应版本的脚本更安全,即 https://github.com/pytorch/vision/tree/v0.10.0/references。