From 3e8c887dfe6cb5f949b31d8a1eadb7deee62fc91 Mon Sep 17 00:00:00 2001 From: Phat Tran <36766404+ptran1203@users.noreply.github.com> Date: Wed, 13 Jan 2021 19:44:31 +0700 Subject: [PATCH] Dynamic shape to support input_shape None dim --- tensorflow/python/keras/applications/resnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/python/keras/applications/resnet.py b/tensorflow/python/keras/applications/resnet.py index 486761df37a..fd9c61ae3f6 100644 --- a/tensorflow/python/keras/applications/resnet.py +++ b/tensorflow/python/keras/applications/resnet.py @@ -405,12 +405,12 @@ def block3(x, depth_multiplier=c, use_bias=False, name=name + '_2_conv')(x) - x_shape = backend.int_shape(x)[1:-1] - x = layers.Reshape(x_shape + (groups, c, c))(x) + x_shape = backend.shape(x)[:-1] + x = backend.reshape(x, backend.concatenate([x_shape, (groups, c, c)])) x = layers.Lambda( lambda x: sum(x[:, :, :, :, i] for i in range(c)), name=name + '_2_reduce')(x) - x = layers.Reshape(x_shape + (filters,))(x) + x = backend.reshape(x, backend.concatenate([x_shape, (filters,)])) x = layers.BatchNormalization( axis=bn_axis, epsilon=1.001e-5, name=name + '_2_bn')(x) x = layers.Activation('relu', name=name + '_2_relu')(x)