diff --git a/src/test/java/hex/DeepLearningProstateTest.java b/src/test/java/hex/DeepLearningProstateTest.java
index <HASH>..<HASH> 100644
--- a/src/test/java/hex/DeepLearningProstateTest.java
+++ b/src/test/java/hex/DeepLearningProstateTest.java
@@ -147,6 +147,7 @@ public class DeepLearningProstateTest extends TestUtil {
                                       }
 
                                       model1 = UKV.get(dest_tmp);
+                                      Assert.assertTrue(p.train_samples_per_iteration <= 0 || model1.epoch_counter > epochs || Math.abs(model1.epoch_counter - epochs)/epochs < 0.1);
 
                                       if (n_folds != 0)
                                       // test HTML of cv models
Add assertion that the number of training rows is correct.