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

fix(pu): polish all mlp model and related configs #26

Merged
merged 16 commits into from
May 7, 2023
Merged

Conversation

puyuan1996
Copy link
Collaborator

@puyuan1996 puyuan1996 commented May 7, 2023

  • fix the norm, activation, and initialization in the mlp model
  • add the option res_connection_in_dynamics. If it's True, we add skip connections of the latent state when learning next_latent_state in the dynamics network which can make the learning of next_latent_state more stable and smooth.

image

@puyuan1996 puyuan1996 added bug Something isn't working enhancement New feature or request labels May 7, 2023
@puyuan1996 puyuan1996 merged commit 1d2b8f2 into main May 7, 2023
@puyuan1996 puyuan1996 deleted the dev-mlp branch May 7, 2023 06:42
@puyuan1996 puyuan1996 restored the dev-mlp branch May 7, 2023 06:42
@puyuan1996 puyuan1996 deleted the dev-mlp branch May 7, 2023 06:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant