Minor fixes

cmusphinx · Apr 14, 2017 · 4bb83d8 · 4bb83d8
1 parent e2b4b3b
commit 4bb83d8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/g2p_seq2seq/g2p.py b/g2p_seq2seq/g2p.py
@@ -215,9 +215,9 @@ def train(self):
     current_step, iter_inx, num_epochs_last_impr, max_num_epochs,\
       num_up_trends, num_down_trends = 0, 0, 0, 2, 0, 0
     prev_train_losses, prev_valid_losses, prev_epoch_valid_losses = [], [], []
-    num_iter_cover_train = int(sum(train_bucket_sizes) /
-                               self.params.batch_size /
-                               self.params.steps_per_checkpoint)
+    num_iter_cover_train = max(1, int(sum(train_bucket_sizes) /
+                                      self.params.batch_size /
+                                      self.params.steps_per_checkpoint))
     while (self.params.max_steps == 0
            or self.model.global_step.eval(self.session)
            <= self.params.max_steps):