我正在查看 AI 平台的文档,以及如何运行训练作业我正在查看以下部分:格式化您的配置参数
在那里,展示了如何在 config.yaml 文件中指定您的输入参数。它仅针对 gcloud 特定参数显示。如何将 USER_ARGS 添加到 config.yaml 文件中。
您可以通过在 trainingInput 下添加“args”列表来在配置文件中提供自定义参数,如下所示:
trainingInput:
scaleTier: CUSTOM
masterType: complex_model_m
workerType: complex_model_m
parameterServerType: large_model
workerCount: 9
parameterServerCount: 3
runtimeVersion: '1.14'
pythonVersion: '3.5'
args:
- '--neuralNetLayers=50000'
- '--tileUniverseWithPaperclips=False'
config.yaml 文件是一个 Job 资源。大多数有趣的参数都在 TrainingInput 中——这里记录了完整的 TrainingInput 规范:https : //cloud.google.com/ml-engine/reference/rest/v1/projects.jobs#traininginput
希望有帮助。快乐学习!
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句