Skip to content
Snippets Groups Projects
Commit 4623a148 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Merge branch 'inceptionv1_fix' into 'master'

Fixed issues with TF1 compatibility

See merge request !91
parents 9a0fa0ca da66361d
No related branches found
No related tags found
1 merge request!91Fixed issues with TF1 compatibility
Pipeline #46483 skipped
...@@ -441,7 +441,7 @@ def InceptionResNetV1( ...@@ -441,7 +441,7 @@ def InceptionResNetV1(
Conv2D_BN(32, 3, strides=2, padding="valid", name="Conv2d_1a_3x3"), Conv2D_BN(32, 3, strides=2, padding="valid", name="Conv2d_1a_3x3"),
Conv2D_BN(32, 3, padding="valid", name="Conv2d_2a_3x3"), Conv2D_BN(32, 3, padding="valid", name="Conv2d_2a_3x3"),
Conv2D_BN(64, 3, name="Conv2d_2b_3x3"), Conv2D_BN(64, 3, name="Conv2d_2b_3x3"),
MaxPool2D(3, strides=2, name="MaxPool_3a_3x3"), MaxPool2D(3, strides=2, name="MaxPool_3a_3x3", padding="valid"),
Conv2D_BN(80, 1, padding="valid", name="Conv2d_3b_1x1"), Conv2D_BN(80, 1, padding="valid", name="Conv2d_3b_1x1"),
Conv2D_BN(192, 3, padding="valid", name="Conv2d_4a_3x3"), Conv2D_BN(192, 3, padding="valid", name="Conv2d_4a_3x3"),
Conv2D_BN(256, 3, strides=2, padding="valid", name="Conv2d_4b_3x3"), Conv2D_BN(256, 3, strides=2, padding="valid", name="Conv2d_4b_3x3"),
...@@ -500,7 +500,7 @@ def InceptionResNetV1( ...@@ -500,7 +500,7 @@ def InceptionResNetV1(
) )
# 5x block8 (Inception-ResNet-C block): 8 x 8 x 2080 # 5x block8 (Inception-ResNet-C block): 8 x 8 x 2080
for block_idx in range(1, 5): for block_idx in range(1, 6):
layers.append( layers.append(
InceptionResnetBlock( InceptionResnetBlock(
n_channels=1792, n_channels=1792,
...@@ -517,8 +517,8 @@ def InceptionResNetV1( ...@@ -517,8 +517,8 @@ def InceptionResNetV1(
scale=1.0, scale=1.0,
activation=None, activation=None,
block_type="block8", block_type="block8",
block_idx=5, block_idx=6,
name="block8_5", name="block8_6",
) )
) )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment