refactor: removed checkpointing at step 0.

Modalities · Jul 15, 2024 · 1aa906b · 1aa906b
1 parent 8e79acc
commit 1aa906b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/modalities/gym.py b/src/modalities/gym.py
@@ -74,7 +74,7 @@ def _run_checkpointing(
         checkpoint_saving: CheckpointSaving,
         checkpointing_interval_in_steps: int,
     ):
-        if num_train_steps_done % checkpointing_interval_in_steps == 0:
+        if num_train_steps_done % checkpointing_interval_in_steps == 0 and num_train_steps_done > 0:
             checkpoint_saving.save_checkpoint(
                 num_train_steps_done=num_train_steps_done,
                 evaluation_result=None,  # TODO implement checkpointing based on preceding evaluation results