Jinpeng Zheng

Maybe the `forward` function in `EncoderDecoder` should be ```python def forward(self, src, tgt, src_mask, tgt_mask): memory = self.encode(src, src_mask) res_dec = self.decode(memory, src_mask, tgt, tgt_mask) return self.generator(res_dec) ```

> Hi, I encountered same issues (working on https://github.com/harvardnlp/annotated-transformer/ ) Any updates on that? I'm learning this repository too. And this issue happend when running: ```python train, val, test =...

This repo https://github.com/Newbeeer/L_DMI provided the download link.