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

Fixed issues with TF1 compatibility

parent 9a0fa0ca
No related branches found
No related tags found
1 merge request!91Fixed issues with TF1 compatibility
Pipeline #46326 passed
...@@ -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