Skip to content
This repository was archived by the owner on Aug 1, 2023. It is now read-only.

Commit 0bcfad8

Browse files
halilakinfacebook-github-bot
authored andcommitted
Fix MI-LSTM GPU training extended test failure
Summary: There were 2 test failures. This is the one that happens on GPU machines. Will look at the other test failure independently. There were 2 issues in this test. One is not calling the parent's init with the correct parameters. And the other one was fairseq trainer requiring all gradient receiving parameters to be in the forward pass computation. (https://fburl.com/g6xtyh80) Reviewed By: jhcross Differential Revision: D9990597 fbshipit-source-id: 1b6c94e609502f68e28bd211f95bcb9535f1c40d
1 parent 087a29c commit 0bcfad8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_translate/rnn_cell.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def LSTMCell(input_dim, hidden_dim, **kwargs):
1717

1818
class MILSTMCellBackend(nn.RNNCell):
1919
def __init__(self, input_size, hidden_size, bias=True):
20-
super(nn.RNNCell, self).__init__()
20+
super(MILSTMCellBackend, self).__init__(input_size, hidden_size, bias=False)
2121
self.input_size = input_size
2222
self.hidden_size = hidden_size
2323
self.bias = bias

0 commit comments

Comments
 (0)