From a085aa3884e45a4709ebb7b2f43880944f2c498c Mon Sep 17 00:00:00 2001 From: Dobiasd Date: Tue, 17 Dec 2019 14:14:52 +0100 Subject: [PATCH] Improve error for unsupported BatchNormalization axes --- keras_export/convert_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/keras_export/convert_model.py b/keras_export/convert_model.py index 3d5d1f9f..5a7db5d7 100755 --- a/keras_export/convert_model.py +++ b/keras_export/convert_model.py @@ -354,7 +354,8 @@ def show_batch_normalization_layer(layer): else: assert len(layer.axis) == 1 layer_axis = layer.axis[0] - assert layer_axis == -1 or layer_axis + 1 == len(layer.input_shape) + assert layer_axis == -1 or layer_axis + 1 == len(layer.input_shape), \ + "BatchNormalization only supported on the last tensor axis" moving_mean = K.get_value(layer.moving_mean) moving_variance = K.get_value(layer.moving_variance) result = {}