目录

训练

训练机器学习模型通常需要自定义训练循环和代码。因此,我们没有提供现成的训练循环应用程序。不过,我们确实有一些示例,展示如何构建您的训练应用程序,以及可以用于运行自定义训练应用程序的通用组件。

  1. 训练器示例

  2. 组件最佳实践

  3. 参见 torchx.components 以了解更多关于编写组件的信息

  4. 欲了解有关分布式训练的更多信息,请参阅torchx.components.dist

嵌入式训练脚本

对于简单的应用程序,您可以使用torchx.components.utils.python() 组件将训练脚本作为命令行参数嵌入到Python命令中。这有一定的大小限制,但对于许多较小的应用程序来说是可以工作的。

>>> from torchx.components.utils import python
>>> app = """
... import sys
... print(f"Hello, {sys.argv[0]}")
... """
>>> python("TorchX user", c=app)
AppDef(..., entrypoint='python', ...)

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

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

查看教程

资源

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

查看资源