flax
changed `return_weights` to `sow_weights` for attention layer
#3550
Merged
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Overview
Commits
1
Changes
View On
GitHub
Commits
changed to for attention layer
chiamp
committed
2 years ago
Loading