Skip to content
Snippets Groups Projects
Commit 192cdd05 authored by lukas leufen's avatar lukas leufen
Browse files

fixed layer in conv block

parent a15a62f5
No related branches found
No related tags found
2 merge requests!6updated inception model and data prep class,!5Running CI
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
src/inception_model.py
+ 1
− 1
View file @ 192cdd05
...@@ -124,7 +124,7 @@ class InceptionModelBase: ...@@ -124,7 +124,7 @@ class InceptionModelBase:
tower = pooling(pool_kernel, strides=(1, 1), padding='same', name=block_name+block_type)(input_x) tower = pooling(pool_kernel, strides=(1, 1), padding='same', name=block_name+block_type)(input_x)
# convolution block # 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) tower = self.act(tower, activation, **act_settings)
return tower return tower
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment