Update train.py

Fixes typo in `train.py` docs.
CUNY-CL · May 7, 2024 · 1425735 · 1425735
1 parent ad3c4e9
commit 1425735
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yoyodyne/train.py b/yoyodyne/train.py
@@ -54,7 +54,7 @@ def _get_callbacks(
         patience (int, optional): number of epochs with no
             progress (according to `patience_metric`) before triggering
             early stopping.
-        checkpoint_metric (string, optional): validation metric used to
+        patience_metric (string, optional): validation metric used to
             trigger early stopping.
 
     Returns: