FIX: some dtype and device inconsistent bug for OR-Transformer

ictnlp / OR-NMT

Source Code for ACL2019 paper <Bridging the Gap between Training and Inference for Neural Machine Translation>

41 stars 10 forks source link

Closed nomadlx closed 4 years ago

nomadlx commented 4 years ago

Fixed some bugs that caused OR-Transformer to fail to run, see #9

zhang-wen commented 4 years ago

@nomadlx Thanks for the bug fix!