flax
65dfd465
- MultiHeadAttention only keeps rngs if dropout_rate is positive
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
223 days ago
MultiHeadAttention only keeps rngs if dropout_rate is positive PiperOrigin-RevId: 759468680
References
#4750 - MultiHeadAttention only keeps rngs if dropout_rate is positive
Author
Cristian Garcia
Committer
a-googler
Parents
45ce6ee2
Loading