[Fix] Remove besk ckpt only in master rank (#682)

open-mmlab · Nov 8, 2022 · 0f62a6c · 0f62a6c
1 parent 103ad60
commit 0f62a6c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mmengine/hooks/checkpoint_hook.py b/mmengine/hooks/checkpoint_hook.py
@@ -408,7 +408,9 @@ def _save_best_checkpoint(self, runner, metrics) -> None:
             best_score = key_score
             runner.message_hub.update_info(best_score_key, best_score)
 
-            if best_ckpt_path and self.file_client.isfile(best_ckpt_path):
+            if best_ckpt_path and \
+               self.file_client.isfile(best_ckpt_path) and \
+               is_main_process():
                 self.file_client.remove(best_ckpt_path)
                 runner.logger.info(
                     f'The previous best checkpoint {best_ckpt_path} '