公开(公告)号：US20240428082A1

公开(公告)日：2024-12-26

申请号：US18491604

申请日：2023-10-20

Applicant: Amazon Technologies, Inc.

Inventor： Zhuang Wang ,  Zhen Jia ,  Shuai Zheng ,  Zhen Zhang ,  Xinwei Fu ,  Yida Wang

IPC: G06N3/098

Abstract: A placement plan for training state checkpoints of a machine learning model is generated based at least in part on a number of training servers of a distributed training environment. The plan indicates, with respect to an individual server, one or more other servers at which replicas of training state checkpoints of the individual server are to be stored. During selected periods of one or more training iterations of the model, respective portions of a replica of a training state checkpoint of a first server are transmitted to a second server selected based on the placement plan. After an event causes disruption of the training iterations, one of the checkpoints generated at the first server is retrieved from the second server and used to resume the training iterations.