Skip to content
This repository has been archived by the owner on Aug 18, 2021. It is now read-only.

Fixes to BahdanauAttnDecoderRNN #119

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

anantzoid
Copy link

This PR fixes all the issue with BahdanauAttnDecoderRNN class and makes the NMT trainable with it as well. The changes consist of the following:

  • Syntax fixes (wrong class in super),
  • Removed unused variable max_length,
  • Handled dot product for batch-size=1 using bmm in Attn class for 'dot' and 'general'
  • Tested both the decoders under the section Testing the models

syntax fixes,
removed unused variable max_length
handled dot product for batch-size=1
tested module
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant