Merge pull request #295 from MeetKai/musab-mk-patch-1

Update train.py
MeetKai · Nov 22, 2024 · dc43ea2 · dc43ea2
2 parents 7885358 + 3f36cb5
commit dc43ea2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/functionary/train/train.py b/functionary/train/train.py
@@ -140,6 +140,8 @@ def trainer_save_model_safe(trainer: transformers.Trainer):
 
 
 def train():
+    """Training loop"""
+
     argument_parser = transformers.HfArgumentParser(
         (ModelArguments, DataArguments, TrainingArguments)
     )