Fix setting of BatchNormalization config. (#36)

broadinstitute · Nov 28, 2017 · fc3e120 · fc3e120
1 parent 2a64eac
commit fc3e120
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/keras_resnet/layers/_batch_normalization.py b/keras_resnet/layers/_batch_normalization.py
@@ -17,4 +17,6 @@ def call(self, *args, **kwargs):
         return super(BatchNormalization, self).call(training=(not self.freeze), *args, **kwargs)
 
     def get_config(self):
-        return {'freeze': self.freeze}
+        config = super(BatchNormalization, self).get_config()
+        config.update({'freeze': self.freeze})
+        return config