From 5dff73082ec5f7dbfe2c1ee7c711b6acf3207742 Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Mon, 16 Oct 2023 18:49:49 -0400 Subject: [PATCH] remove flush on close (#2646) --- composer/callbacks/checkpoint_saver.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/composer/callbacks/checkpoint_saver.py b/composer/callbacks/checkpoint_saver.py index 17da58236d..6b81cda7ab 100644 --- a/composer/callbacks/checkpoint_saver.py +++ b/composer/callbacks/checkpoint_saver.py @@ -304,14 +304,6 @@ def epoch_checkpoint(self, state: State, logger: Logger): logger, ) - def close(self, state: State, logger: Logger): - trained_at_least_one_batch = self.start_batch is not None and self.start_batch != state.timestamp.batch - if self.last_checkpoint_batch != state.timestamp.batch and trained_at_least_one_batch: - self._save_checkpoint( - state, - logger, - ) - def get_state_dict(self, state): return { 'state': state.state_dict(),