Update trainer for modified Optimizer

explosion · Oct 28, 2017 · c1cad42 · c1cad42
1 parent d2c7e94
commit c1cad42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thinc/neural/train.py b/thinc/neural/train.py
@@ -40,7 +40,7 @@ def iterate(self, train_X, train_y, progress_bar=True):
                     y = _take_slice(train_y, slice_)
                     yield X, y
                     self.dropout = linear_decay(orig_dropout, self.dropout_decay,
-                                                self.optimizer.nr_iter)
+                                                j)
                     j += self.batch_size
                     if progress_bar:
                         pbar.update(self.batch_size)