如何让HuggingFace SageMaker训练任务后台运行以关闭本地电脑？

• 默认情况下，huggingface_estimator.fit()方法的wait参数为True，会阻塞本地进程直到训练完成，这就是关闭本地电脑后训练停止的原因。
• 只需在调用fit时添加wait=False参数，即可让训练任务在SageMaker云端后台异步运行——提交任务后本地进程直接结束，关闭电脑完全不影响云端的训练进程。

修改后的代码示例：

huggingface_estimator.fit(
  {
    'train': 's3://sagemaker-us-east-1-558105141721/samples/datasets/imdb/train',
    'test': 's3://sagemaker-us-east-1-558105141721/samples/datasets/imdb/test'
  },
  wait=False
)

后续查看训练状态的方式 #

• 调用huggingface_estimator.describe()可获取训练任务的详细状态信息
• 直接登录AWS SageMaker控制台，在「训练任务」页面查看实时进度与日志

内容的提问来源于stack exchange，提问作者lmaooooo