Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

persist Seq2Seq through keras load - save api #224

Open
wants to merge 5 commits into
base: master
Choose a base branch
from

Conversation

stefanoterna
Copy link

Added a get_config override method to handle hidden_dim in LSTMDecoderCell and AttentionDecoderCell.

Moreover, the constructors of both classes relied on self.output_dim to set the value of hidden_dim if not supplied. But self.output_dim is not available until super constructor has been called. Fixed by accessing directly to kwargs, so that hidden_dim is be available when the super constructor calls the build_model method.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant