d2l-en icon indicating copy to clipboard operation
d2l-en copied to clipboard

Need to tune performance for MXNet & TensorFlow for seq2seq

Open astonzhang opened this issue 2 years ago • 1 comments

http://preview.d2l.ai.s3-website-us-west-2.amazonaws.com/d2l-en/master/chapter_recurrent-modern/seq2seq.html http://preview.d2l.ai.s3-website-us-west-2.amazonaws.com/d2l-en/master/chapter_attention-mechanisms/bahdanau-attention.html http://preview.d2l.ai.s3-website-us-west-2.amazonaws.com/d2l-en/master/chapter_attention-mechanisms/transformer.html

We need to tune performance for MXNet & TensorFlow to obtain similar performance of PyTorch for each section, such as learning rate & max_epochs.

astonzhang avatar Apr 13 '22 07:04 astonzhang

http://preview.d2l.ai.s3-website-us-west-2.amazonaws.com/d2l-en/master/chapter_attention-mechanisms/bahdanau-attention.html was fixed in #2104. seq2seq and transformers for TF still need fixing.

AnirudhDagar avatar May 16 '22 13:05 AnirudhDagar