训练参考¶
除了许多模型、数据集和图像变换之外,Torchvision 还提供了训练参考脚本。这些是我们用来训练 模型 的脚本,这些模型随后可以使用预训练权重提供。
这些脚本不是核心包的一部分,而是可以在GitHub上获得。我们目前提供以下参考:分类、检测、分割、相似度学习和视频分类。
尽管这些脚本在很大程度上是稳定的,但它们不提供向后兼容性保证。
通常,这些脚本依赖于最新(尚未发布的)PyTorch版本 或最新版的torchvision。这意味着要使用它们,你可能需要安装最新版的PyTorch和torchvision,例如:
conda install pytorch torchvision -c pytorch-nightly
如果你需要依赖较旧的稳定版本的 PyTorch 或 torchvision,例如 torchvision 0.10,则使用该相应版本在 GitHub 上的脚本更安全,即 https://github.com/pytorch/vision/tree/v0.10.0/references。