(self, encoder, decoder)
| 1122 | |
| 1123 | class GreedySearchDecoder(nn.Module): |
| 1124 | def __init__(self, encoder, decoder): |
| 1125 | super(GreedySearchDecoder, self).__init__() |
| 1126 | self.encoder = encoder |
| 1127 | self.decoder = decoder |
| 1128 | |
| 1129 | def forward(self, input_seq, input_length, max_length): |
| 1130 | # Forward input through encoder model |