flax
1c6cb6b0 - set axis='batch'

Commit
2 years ago
set axis='batch' Without this the accuracy is 71.6 on 4x4. Adding this the accuracy is 76.5 on 4x4.
Author
Parents
Loading