diff --git a/src/model_modules/inception_model.py b/src/model_modules/inception_model.py index 11093df56a4b262f75eae7a6f7c05e6e44e1435d..1daddd84ede9131718e4ba4354fba5157270f2dc 100644 --- a/src/model_modules/inception_model.py +++ b/src/model_modules/inception_model.py @@ -153,6 +153,7 @@ class InceptionModelBase: self.number_of_blocks += 1 self.part_of_block = 0 tower_build = {} + block_name = f"Block_{self.number_of_blocks}" for part, part_settings in tower_conv_parts.items(): tower_build[part] = self.create_conv_tower(input_x, **part_settings, **kwargs) if 'max_pooling' in tower_pool_parts.keys(): @@ -165,7 +166,8 @@ class InceptionModelBase: tower_build['maxpool'] = self.create_pool_tower(input_x, **tower_pool_parts, **kwargs) tower_build['avgpool'] = self.create_pool_tower(input_x, **tower_pool_parts, **kwargs, max_pooling=False) - block = keras.layers.concatenate(list(tower_build.values()), axis=3) + block = keras.layers.concatenate(list(tower_build.values()), axis=3, + name=block_name+"_Co") return block