search

broadinstitute / keras-resnet

Keras package for deep residual networks
Other
300 stars 127 forks source link

Fix setting of BatchNormalization config. #36

Closed hgaiser closed 6 years ago

hgaiser commented 6 years ago

The configuration of the batch normalization layers wasn't stored correctly, because the super class was not called. This causes parameters like name or epsilon to be lost when loading a previously saved model.

0x00b1 commented 6 years ago

Thanks, @hgaiser!

0x00b1 commented 6 years ago

Thanks, @hgaiser!