[BART] Remove unused kwargs (#3279)

* Remove unused kwargs * dont call forward in tests
2020-03-15 23:00:44 -04:00
parent 3814e167d9
commit 5ea8ba67b4
3 changed files with 14 additions and 29 deletions
--- a/examples/summarization/bertabs/modeling_bertabs.py
+++ b/examples/summarization/bertabs/modeling_bertabs.py
@@ -844,7 +844,7 @@ class Translator(object):
            dec_out, dec_states = self.model.decoder(decoder_input, src_features, dec_states, step=step)

            # Generator forward.
-            log_probs = self.generator.forward(dec_out.transpose(0, 1).squeeze(0))
+            log_probs = self.generator(dec_out.transpose(0, 1).squeeze(0))
            vocab_size = log_probs.size(-1)

            if step < min_length: