Fix DMP state dict bug when wrapped with lightning

facebookresearch / recipes

Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research techniques without significant engineering overhead.Specifically, recipes aims to provide- Consistent access to pre-trained SOTA models ready for production- Reference implementations for SOTA research reproducibility, and infrastructure to guarantee correctness, efficiency, and interoperability.

BSD 3-Clause "New" or "Revised" License

284 stars 35 forks source link

Fix DMP state dict bug when wrapped with lightning #20

Closed joshuadeng closed 2 years ago

joshuadeng commented 2 years ago

Summary: The state dict was empty when calling state dict method from a lightning module wrapping DMP. The destination dict was not being passed through.

Differential Revision: D35783627

facebook-github-bot commented 2 years ago

This pull request was exported from Phabricator. Differential Revision: D35783627