[FIX] Houglass Layer recursive call

wbenbihi · Aug 21, 2022 · 2c2aaa9 · 2c2aaa9
1 parent 28997b2
commit 2c2aaa9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hourglass_tensorflow/layers/hourglass.py b/hourglass_tensorflow/layers/hourglass.py
@@ -78,7 +78,7 @@ def __init__(
                 dynamic=dynamic,
                 trainable=trainable,
             )
-            downsampling["low_3"] = layers.UpSampling2D(
+            downsampling["up_2"] = layers.UpSampling2D(
                 size=(2, 2),
                 data_format=None,
                 interpolation="nearest",
@@ -101,7 +101,7 @@ def _recursive_call(self, input_tensor, step, training=True):
         low_ = step_layers["low_"](input_tensor, training=training)
         low_1 = step_layers["low_1"](low_, training=training)
         if step == 0:
-            low_2 = step_layers["low_2"](low_, training=training)
+            low_2 = step_layers["low_2"](low_1, training=training)
         else:
             low_2 = self._recursive_call(low_1, step=(step - 1), training=training)
         low_3 = step_layers["low_3"](low_2, training=training)