[BART] Remove unused kwargs (#3279)
* Remove unused kwargs * dont call forward in tests
This commit is contained in:
1 parent
3814e167d9
commit
5ea8ba67b4
3 files changed
+14
-29
No files matched your search
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user