flax
e4c00a76 - Merge pull request #137 from google/batchnorm-singletons

Commit

6 years ago

Merge pull request #137 from google/batchnorm-singletons Avoid singleton dimensions in batchnorm params

References

Author

avital

Parents