()
| 15 | |
| 16 | # Simple test to show how our network works |
| 17 | def test(): |
| 18 | encoder_hidden = encoder.init_hidden() |
| 19 | word_input = str2tensor('hello') |
| 20 | encoder_outputs, encoder_hidden = encoder(word_input, encoder_hidden) |
| 21 | print(encoder_outputs) |
| 22 | |
| 23 | decoder_hidden = encoder_hidden |
| 24 | |
| 25 | word_target = str2tensor('pytorch') |
| 26 | for c in range(len(word_target)): |
| 27 | decoder_output, decoder_hidden = decoder( |
| 28 | word_target[c], decoder_hidden) |
| 29 | print(decoder_output.size(), decoder_hidden.size()) |
| 30 | |
| 31 | |
| 32 | # Train for a given src and target |
no test coverage detected