pytorch
786903ea - Provide an auto wrap policy for common transformer models

Commit
2 years ago
Provide an auto wrap policy for common transformer models Pull Request resolved: https://github.com/pytorch/pytorch/pull/76455 Provide an auto wrap policy for common transformer models Differential Revision: [D35972488](https://our.internmc.facebook.com/intern/diff/D35972488/) Approved by: https://github.com/rohan-varma
Author
Committer
Parents
Loading