From b94e951633577317b4b029836d0bb0cf05b121d6 Mon Sep 17 00:00:00 2001 From: Aadyot Bhatnagar Date: Wed, 27 Aug 2025 22:24:36 +0000 Subject: [PATCH] Fix autoresume with non-default CheckpointSaver. --- composer/trainer/trainer.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/composer/trainer/trainer.py b/composer/trainer/trainer.py index cfe22067c9..5a17e1a264 100644 --- a/composer/trainer/trainer.py +++ b/composer/trainer/trainer.py @@ -1375,9 +1375,13 @@ def __init__( log.info('Multiple CheckpointSaver provided as callbacks. Using the first one as reference.') self._checkpoint_saver = _checkpoint_savers[0] - if self._checkpoint_saver.folder != save_folder: - log.info(f'Using {self._checkpoint_saver.folder} as save_folder.') - save_folder = self._checkpoint_saver.folder + if self._checkpoint_saver.remote_uploader: + checkpoint_saver_folder = self._checkpoint_saver.remote_uploader.remote_folder + else: + checkpoint_saver_folder = self._checkpoint_saver.folder + if checkpoint_saver_folder != save_folder: + log.info(f'Using {checkpoint_saver_folder} as save_folder.') + save_folder = checkpoint_saver_folder if self._checkpoint_saver.latest_filename is None: save_latest_filename = None