From 73c50f77920aead4e8c71dba387822cdbeb16fd7 Mon Sep 17 00:00:00 2001 From: Christian Bodenstein Date: Sat, 25 Jul 2020 03:14:38 +0200 Subject: [PATCH] fix saved model serialization --- keras_resnet/layers/_batch_normalization.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/keras_resnet/layers/_batch_normalization.py b/keras_resnet/layers/_batch_normalization.py index 1946a3f..b2c08cb 100644 --- a/keras_resnet/layers/_batch_normalization.py +++ b/keras_resnet/layers/_batch_normalization.py @@ -12,11 +12,11 @@ def __init__(self, freeze, *args, **kwargs): # set to non-trainable if freeze is true self.trainable = not self.freeze - def call(self, *args, **kwargs): + def call(self, inputs, **kwargs): # Force test mode if frozen, otherwise use default behaviour (i.e., training=None). if self.freeze: kwargs['training'] = False - return super(BatchNormalization, self).call(*args, **kwargs) + return super(BatchNormalization, self).call(inputs, **kwargs) def get_config(self): config = super(BatchNormalization, self).get_config()