(self, config)
| 54 | return self.encoder |
| 55 | |
| 56 | def init_decoder(self, config): |
| 57 | self.decoder = Decoder(config) |
| 58 | self.decoder.blocks.ssm.lag = self.lag |
| 59 | self.decoder.blocks.ssm.horizon = self.horizon |
| 60 | return self.decoder |
| 61 | |
| 62 | def init_output(self, config): |
| 63 | return init_mlp(config) |