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: update ret_output shape #1147

Merged
merged 3 commits into from
Dec 3, 2024
Merged

Conversation

ArnolFokam
Copy link
Contributor

Closes #1146

@CLAassistant
Copy link

CLAassistant commented Nov 29, 2024

CLA assistant check
All committers have signed the CLA.

sash-a
sash-a previously approved these changes Nov 29, 2024
Copy link
Contributor

@sash-a sash-a left a comment

Choose a reason for hiding this comment

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

Great spot, thanks @ArnolFokam 🔥

Copy link
Contributor

@OmaymaMahjoub OmaymaMahjoub left a comment

Choose a reason for hiding this comment

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

Thanks @ArnolFokam 🔥

Copy link
Contributor

@sash-a sash-a left a comment

Choose a reason for hiding this comment

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

🙏

@sash-a sash-a merged commit 2762b3d into develop Dec 3, 2024
4 checks passed
@sash-a sash-a deleted the fix/retention-heads-output-sable branch December 3, 2024 07:36
sash-a pushed a commit that referenced this pull request Dec 13, 2024
* fix: update ret_output shape

* fix: update w_g

* fix: update w_o shape
sash-a added a commit that referenced this pull request Jan 10, 2025
- chore: remove jax.tree_map
- Merge branch develop into chore/ppo-system-cleanup
- fix: update ret_output shape (#1147)
- Merge branch develop into chore/ppo-system-cleanup
- chore: minor comment changes
- fix: integration test workflow (#1145)
- Merge branch develop into chore/ppo-system-cleanup
- Merge branch develop into chore/ppo-system-cleanup
- chore: renaming loss_X -> X_loss
- fix: updated sebulba
- Merge branch develop into chore/ppo-system-cleanup
- chore: remove expired link to anakin notebook
- chore: add back advanced usage in examples
- Merge branch develop into chore/ppo-system-cleanup
- feat: sable cleanup
- chore: merge dev
- chore: unify value/critic loss naming
- refactor: loss_actor -> actor_loss
- Merge branch develop into chore/ppo-system-cleanup
- chore: pre-commit
- feat: mat system clean up
- chore: remove advanced usage
- Merge branch develop into chore/ppo-system-cleanup
- feat: ff mappo system clean up
- feat: ff ippo system clean up
- feat: rec mappo system clean up
- feat: rec ippo system clean up

Co-authored-by: Ruan de Kock <[email protected]>
Co-authored-by: Louay-Ben-nessir <[email protected]>
Co-authored-by: Arnol Fokam <[email protected]>
sash-a added a commit that referenced this pull request Jan 10, 2025
- chore: remove jax.tree_map
- Merge branch develop into chore/ppo-system-cleanup
- fix: update ret_output shape (#1147)
- Merge branch develop into chore/ppo-system-cleanup
- chore: minor comment changes
- fix: integration test workflow (#1145)
- Merge branch develop into chore/ppo-system-cleanup
- Merge branch develop into chore/ppo-system-cleanup
- chore: renaming loss_X -> X_loss
- fix: updated sebulba
- Merge branch develop into chore/ppo-system-cleanup
- chore: remove expired link to anakin notebook
- chore: add back advanced usage in examples
- Merge branch develop into chore/ppo-system-cleanup
- feat: sable cleanup
- chore: merge dev
- chore: unify value/critic loss naming
- refactor: loss_actor -> actor_loss
- Merge branch develop into chore/ppo-system-cleanup
- chore: pre-commit
- feat: mat system clean up
- chore: remove advanced usage
- Merge branch develop into chore/ppo-system-cleanup
- feat: ff mappo system clean up
- feat: ff ippo system clean up
- feat: rec mappo system clean up
- feat: rec ippo system clean up

Co-authored-by: Louay-Ben-nessir <[email protected]>
Co-authored-by: Arnol Fokam <[email protected]>
Co-authored-by: Ruan de Kock <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[BUG] output from multi-scale retention only contains output from the first head
4 participants