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
Branches
Tags
2 merge requests!6updated inception model and data prep class,!5Running CI
Show 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.
Please register or to comment