flax
7f10ad8d
- changed to for attention layer
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
changed to for attention layer
References
#3550 - changed `return_weights` to `sow_weights` for attention layer
Author
chiamp
Parents
2e51a4e5
Loading