t2 random

broadinstitute · Mar 4, 2025 · 2fdb3a3 · 2fdb3a3
1 parent dd11f56
commit 2fdb3a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4h/plots.py b/ml4h/plots.py
@@ -463,7 +463,7 @@ def plot_metric_history(history, training_steps: int, title: str, prefix="./figu
                 col += 1
                 if col >= cols:
                     break
-        if len(history.history[k]) < 1:
+        if len(history.history[k]) < 2:
             logging.info(f"Not enough epochs to plot learning curves at:{k}")
             return