Skip to content

Conversation

mattdangerw
Copy link
Member

No description provided.

Copy link
Contributor

@chenmoneygithub chenmoneygithub left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! dropped one minor comment.

x = self.out_proj(x)
return x

def get_config(self):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why aare we deleting the get_config here?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The head is never saved, and we don't want to expose it as a component to other scripts. So I think this does nothing?

Checkpointing doesn't need it. Let's keep things minimal.

@mattdangerw mattdangerw merged commit ec834c3 into keras-team:master May 19, 2022
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.

2 participants