(self, x, seq_lengths)
| 119 | self.optimizer = opt.SGD(0.1) |
| 120 | |
| 121 | def forward(self, x, seq_lengths): |
| 122 | y = self.lstm(x, seq_lengths=seq_lengths) |
| 123 | y = autograd.reshape(y, (y.shape[0], -1)) |
| 124 | y = self.l1(y) |
| 125 | return y |
| 126 | |
| 127 | |
| 128 | class LSTMModel2(model.Model): |
no test coverage detected