目录

火车

训练机器学习模型通常需要自定义训练循环和自定义 法典。因此,我们不提供开箱即用的 Training Loop 应用程序。我们这样做 但是,请通过示例来说明如何构建训练 App 以及 可用于运行自定义训练应用程序的通用组件。

  1. Trainer 应用程序示例

  2. Trainer 组件

  3. 组件最佳实践

  4. torchx.components了解有关编写组件的更多信息

  5. 有关分布式训练的更多信息,请参阅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 的全面开发人员文档

查看文档

教程

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

查看教程

资源

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

查看资源