fix config

Author: ankitageorge

recipes/configs/llama3_2/3B_lora.yaml

@@ -77,7 +77,7 @@ loss:
   _component_: torchtune.modules.loss.LinearCrossEntropyLoss
 
 # Training
-epochs: 2
+epochs: 1
 max_steps_per_epoch: null
 gradient_accumulation_steps: 8  # Use to increase effective batch size
 clip_grad_norm: null

torchtune/training/checkpointing/_checkpoint_client.py

@@ -365,6 +365,7 @@ def save_checkpoint(
         checkpointer user has configured.
         """
         intermediate_checkpoint = epoch + 1 < training_progress.total_epochs
+
         if intermediate_checkpoint and self._enable_async_checkpointing:
             self._save_checkpoint_async(
                 model,