flax
43d022a8
- [linen] add individual param initializer in MultiHeadDotProductAttention
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[linen] add individual param initializer in MultiHeadDotProductAttention
References
#3835 - [linen] enable separate initializers for out layer in MultiHeadDotProductAttention
Author
cgarciae
Committer
cgarciae
Parents
27184559
Loading