火车¶
训练机器学习模型通常需要自定义训练循环和自定义 法典。因此,我们不提供开箱即用的 Training Loop 应用程序。我们这样做 但是,请通过示例来说明如何构建训练 App 以及 可用于运行自定义训练应用程序的通用组件。
看
torchx.components
了解有关编写组件的更多信息有关分布式训练的更多信息,请参阅
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', ...)