如何分离 HuggingFace SageMaker 训练？

Question

我正在通过 SageMaker 集成 远程训练 HuggingFace 模型。我的训练工作需要两个多小时，我想在训练期间关闭电脑。

我使用以下代码片段来训练模型：

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'
  }
)

如何设置训练器，使其训练过程在后台运行，以便我可以关闭计算机？

Answer 1

如何设置训练器，使其训练过程在后台运行，以便我可以关闭计算机？

你可以！

在拥抱面部估计器上调用

fit

开始训练工作。一旦训练作业在 SageMaker 上启动，它就会独立于您的本地计算机运行。

您可以关闭机器，作业将继续在 SageMaker 上运行，直到完成。