From da66361d53f19eb8794980c7d4b1ae4f471b3925 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Fri, 4 Dec 2020 18:42:38 +0100
Subject: [PATCH] Fixed issues with TF1 compatibility
---
bob/learn/tensorflow/models/inception_resnet_v1.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/bob/learn/tensorflow/models/inception_resnet_v1.py b/bob/learn/tensorflow/models/inception_resnet_v1.py
index bfab04e1..5e85f21d 100644
--- a/bob/learn/tensorflow/models/inception_resnet_v1.py
+++ b/bob/learn/tensorflow/models/inception_resnet_v1.py
@@ -441,7 +441,7 @@ def InceptionResNetV1(
Conv2D_BN(32, 3, strides=2, padding="valid", name="Conv2d_1a_3x3"),
Conv2D_BN(32, 3, padding="valid", name="Conv2d_2a_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(192, 3, padding="valid", name="Conv2d_4a_3x3"),
Conv2D_BN(256, 3, strides=2, padding="valid", name="Conv2d_4b_3x3"),
@@ -500,7 +500,7 @@ def InceptionResNetV1(
)
# 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(
InceptionResnetBlock(
n_channels=1792,
@@ -517,8 +517,8 @@ def InceptionResNetV1(
scale=1.0,
activation=None,
block_type="block8",
- block_idx=5,
- name="block8_5",
+ block_idx=6,
+ name="block8_6",
)
)
--
GitLab