flax
d3b2cbb2
- Allow reduction axes in GroupNorm to be defined explicitly at layer construction
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Allow reduction axes in GroupNorm to be defined explicitly at layer construction PiperOrigin-RevId: 586995099
Author
a-googler
Committer
a-googler
Parents
c0a2657a
Loading
flax
d3b2cbb2 - Allow reduction axes in GroupNorm to be defined explicitly at layer construction
