GH-3429: Rename parameter group

flairNLP · Oct 29, 2024 · eb6f5b0 · eb6f5b0
1 parent 80457dc
commit eb6f5b0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/flair/trainers/trainer.py b/flair/trainers/trainer.py
@@ -167,7 +167,7 @@ def train(
         create_file_logs: bool = True,
         create_loss_file: bool = True,
         write_weights: bool = False,
-        # scaling
+        # acceleration
         multi_gpu: bool = False,
         # plugins
         plugins: Optional[List[TrainerPlugin]] = None,
@@ -243,7 +243,7 @@ def fine_tune(
         create_file_logs: bool = True,
         create_loss_file: bool = True,
         write_weights: bool = False,
-        # scaling
+        # acceleration
         use_amp: bool = False,
         multi_gpu: bool = False,
         # plugins
@@ -294,7 +294,7 @@ def fine_tune(
             create_file_logs=create_file_logs,
             create_loss_file=create_loss_file,
             write_weights=write_weights,
-            # scaling
+            # acceleration
             use_amp=use_amp,
             multi_gpu=multi_gpu,
             # plugins
@@ -339,7 +339,7 @@ def train_custom(
         create_file_logs: bool = True,
         create_loss_file: bool = True,
         write_weights: bool = False,
-        # scaling
+        # acceleration
         use_amp: bool = False,
         multi_gpu: bool = False,
         # plugins