diff --git a/train.py b/train.py index e8eab90..262f3c0 100644 --- 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