flax
a1090ef5 - Add implementation as linen for imagenet benchmark. Also add epochs=2 as parameters.
Commit
5 years ago
References
Author
Mohit Reddy
Committer
Parents