a

mosaicml · Jun 14, 2024 · ca46b4f · ca46b4f
1 parent 2c77da9
commit ca46b4f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/composer/callbacks/checkpoint_saver.py b/composer/callbacks/checkpoint_saver.py
@@ -469,7 +469,6 @@ def _save_checkpoint(self, state: State, logger: Logger, wait_previous_remote_up
         # Wait the previous upload tasks on all ranks
         # self.wait() has dist.barrier, so it needs to be called
         # on all ranks before any early return
-        #if wait_previous_remote_upload_tasks and self.count / self.num_concurrent_uploads == 0:
         if wait_previous_remote_upload_tasks:
             self.wait()