Add warning about early stopping not working

wellcometrust · Feb 24, 2021 · f6d5eda · f6d5eda
1 parent b10e2a5
commit f6d5eda
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/wellcomeml/ml/cnn.py b/wellcomeml/ml/cnn.py
@@ -271,7 +271,9 @@ def fit(self, X, Y=None, embedding_matrix=None, steps_per_epoch=None):
         if train_steps_per_epoch == 0:
             logger.warning(
                 "Not enough data for validation. Consider decreasing \
-                batch_size or validation_split"
+                batch_size or validation_split. Some features that \
+                rely on validation metrics like early stopping might \
+                not work"
             )
         else:
             steps_per_epoch = train_steps_per_epoch