diff --git a/src/inception_model.py b/src/inception_model.py index 09e032ca70a8e63d9459464e968d25fac29cd4f1..64b5e09674dd9fccd8282b80c4fb9e87bbd5ea5c 100644 --- a/src/inception_model.py +++ b/src/inception_model.py @@ -124,7 +124,7 @@ class InceptionModelBase: tower = pooling(pool_kernel, strides=(1, 1), padding='same', name=block_name+block_type)(input_x) # convolution block - tower = Conv2D(tower_filter, (1, 1), padding='same', name=block_name+"1x1")(tower) + tower = layers.Conv2D(tower_filter, (1, 1), padding='same', name=block_name+"1x1")(tower) tower = self.act(tower, activation, **act_settings) return tower