培训参考¶
除了许多模型、数据集和图像转换之外,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。