Shortcuts

培训参考资料

除了许多模型、数据集和图像变换之外,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