flax

4c2319fd - Removed a superfluous assignment in the examples/imagenet/train.py source.