ypeleg / nfnets-keras

Keras implementation of Normalizer-Free Networks and SGD - Adaptive Gradient Clipping
MIT License
68 stars 21 forks source link

Update nfnet.py #13

Open Jerry-Master opened 3 years ago

Jerry-Master commented 3 years ago

You use the value of self.variant before initializing it. A simple change in line 89 fixes it.