LossCompute: remove epoch arg

epoch arg in LossCompute constructor isn't used, remove it.
OpenNMT · Sep 19, 2017 · b7e653d · b7e653d
1 parent c49d948
commit b7e653d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/onmt/Loss.py b/onmt/Loss.py
@@ -79,12 +79,11 @@ def shards(state, shard_size, eval=False):
 
 
 class LossCompute(object):
-    def __init__(self, generator, crit, tgt_vocab, dataset, epoch, copy_attn):
+    def __init__(self, generator, crit, tgt_vocab, dataset, copy_attn):
         self.generator = generator
         self.crit = crit
         self.tgt_vocab = tgt_vocab
         self.dataset = dataset
-        self.epoch = epoch
         self.copy_attn = copy_attn
 
     def make_loss_batch(self, outputs, batch, attns, range_):

diff --git a/onmt/Trainer.py b/onmt/Trainer.py
@@ -38,7 +38,7 @@ def train(self, epoch, report_func=None):
         """ Called for each epoch to train. """
         closs = onmt.Loss.LossCompute(self.model.generator, self.criterion,
                                       self.fields["tgt"].vocab,
-                                      self.train_data, epoch,
+                                      self.train_data,
                                       self.copy_attn)
 
         total_stats = onmt.Statistics()
@@ -109,7 +109,7 @@ def validate(self):
 
         loss = onmt.Loss.LossCompute(self.model.generator, self.criterion,
                                      self.fields["tgt"].vocab,
-                                     self.valid_data, 0,
+                                     self.valid_data,
                                      self.copy_attn)
         stats = onmt.Statistics()