快捷方式

训练参考

除了提供众多模型、数据集和图像变换外,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 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

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

查看资源