Skip to content
This repository was archived by the owner on Aug 7, 2024. It is now read-only.

Conversation

drisspg
Copy link
Contributor

@drisspg drisspg commented Jan 17, 2024

Summary

Fix test/test_fsdp.py to match correct state dict entries

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jan 17, 2024
@facebook-github-bot
Copy link
Contributor

@drisspg has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

Copy link

@awgu awgu left a comment

Choose a reason for hiding this comment

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

Nice!

Will we always compile FSDP at the root module? How are other places handling this _orig_mod. prefix?

@drisspg
Copy link
Contributor Author

drisspg commented Jan 17, 2024

@awgu this is our only test doing the state dict matching, so I think this is the only place where this error comes up

@facebook-github-bot
Copy link
Contributor

@drisspg merged this pull request in 713d2db.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants