快捷方式

训练参考

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源