Fixes #187.

persephone-tools · Oct 13, 2018 · 0926356 · 0926356
1 parent a216707
commit 0926356
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/persephone/model.py b/persephone/model.py
@@ -407,6 +407,9 @@ def train(self, early_stopping_steps: int = 10, min_epochs: int = 30,
                                         feed_dict=feed_dict)
 
                         train_ler_total += ler
+                    else:
+                        raise PersephoneError("No training data was provided."
+                                              " Check your batch generation.")
 
                     feed_dict = {self.batch_x: valid_x,
                                 self.batch_x_lens: valid_x_lens,