accelerate
102caf4f
- bugfix in swapping init module weights (#2305)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
bugfix in swapping init module weights (#2305) Signed-off-by: Sudhakar Singh <sudhakars@nvidia.com>
References
#2305 - Fixes bug in swapping weights when replacing with Transformer-Engine layers
Author
sudhakarsingh27
Parents
07df5d26
Loading