Migrate transformer_auto_wrap_policy to ModuleWrapPolicy

facebookresearch / d2go

D2Go is a toolkit for efficient deep learning

Apache License 2.0

826 stars 197 forks source link

Closed YanjunChen329 closed 8 months ago

YanjunChen329 commented 8 months ago

Summary: transformer_auto_wrap_policy is buggy and causes issues when wrapping wrapped module. Migrate to ModuleWrapPolicy

Reviewed By: tglik

Differential Revision: D51124721

facebook-github-bot commented 8 months ago

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

facebook-github-bot commented 8 months ago

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

facebook-github-bot commented 8 months ago

This pull request has been merged in facebookresearch/d2go@40e781537dc6b80996d3222e498207f65ccacf6b.