为 AWS SageMaker 培训作业分配名称

Question

这可能是一个非常基本的问题（我想是）

estimator = PyTorch(entry_point='train.py',
                   source_dir = 'code',
                    role = role,
                   framework_version = '1.5.0',
                   py_version = 'py3',
                   instance_count = 2,
                   instance_type = 'ml.g4dn.2xlarge',
                   hyperparameters={"epochs": 2,
                                     "num_labels": 46,
                                     "backend": "gloo",    
                                    },
                   profiler_config=profiler_config,
                    debugger_hook_config=debugger_hook_config,
                    rules=rules
                   )

我如上所述声明我的估算器，并使用 fit() 将其放入训练中。
我已经在我的 sagemaker 上完成了其中的几个，并且在 aws 培训工作日志中有几个培训工作。
但是它们都以'pytorch-training-2021 ....'的形式出现。
无论如何我可以像 'custom-model-xgboost-ver1' 这样声明培训工作的名称吗？
我以为它可以作为估算器的参数之一，但我找不到它。

提前致谢。

Answer 1

当你调用fit()时你可以传递这个参数job_name=yourJobName

Answer 2

您可以使用以下代码段作为示例，使用 estimator.fit()

from sagemaker.pytorch import PyTorch

estimator = PyTorch(entry_point='mnist.py',
                    role=role,
                    py_version='py3',
                    framework_version='1.8.0',
                    instance_count=2,
                    instance_type='ml.c5.2xlarge',
                    hyperparameters={
                        'epochs': 1,
                        'backend': 'gloo'
                    })
estimator.fit('training': inputs, job_name='Name')

为 AWS SageMaker 培训作业分配名称

Assigning name to AWS SageMaker Training job

amazon-web-services

amazon-sagemaker