From 192cdd05d0cdd55de8e79afddc2ad379e58d63f1 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Wed, 6 Nov 2019 13:29:49 +0100
Subject: [PATCH] fixed layer in conv block
---
src/inception_model.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/inception_model.py b/src/inception_model.py
index 09e032ca..64b5e096 100644
--- a/src/inception_model.py
+++ b/src/inception_model.py
@@ -124,7 +124,7 @@ class InceptionModelBase:
tower = pooling(pool_kernel, strides=(1, 1), padding='same', name=block_name+block_type)(input_x)
# 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)
return tower
--
GitLab