Remove eval_frequency as it is better to evaluate constantly

jshuadvd · Jul 8, 2024 · f03ea61 · f03ea61
1 parent ebb16dd
commit f03ea61
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -140,7 +140,6 @@ def train(
     epochs=10,
     gradient_accumulation_steps=4,
     resume_from_checkpoint=None,
-    eval_frequency=5,
 ):
     """
     Train the LongRoPE model.
@@ -156,7 +155,6 @@ def train(
         epochs (int): Number of training epochs.
         gradient_accumulation_steps (int): Number of steps to accumulate gradients.
         resume_from_checkpoint (str): Path to a checkpoint to resume training from.
-        eval_frequency (int): Number of epochs between evaluations.
 
     Returns:
         None