訓練¶
訓練機器學習模型通常需要自訂訓練迴圈和自訂程式碼。因此,我們沒有提供開箱即用的訓練迴圈應用程式。但是,我們確實提供了一些範例,說明如何建構訓練應用程式,以及可以用於執行自訂訓練應用程式的通用元件。
請參閱
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', ...)