add training args to downsample conv to avoid possible low level bug

edwardyehuang · Feb 15, 2022 · 54ce0f9 · 54ce0f9
1 parent 71a06b8
commit 54ce0f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backbones/hrnet.py b/backbones/hrnet.py
@@ -41,7 +41,7 @@ def call(self, inputs, training=None):
         residual = tf.identity(inputs, name="residual")
 
         if self.downsample is not None:
-            residual = self.downsample(residual)
+            residual = self.downsample(residual, training=training)
 
         x = self.conv1(inputs)
         x = self.bn1(x, training=training)
@@ -85,7 +85,7 @@ def call(self, inputs, training=None):
         residual = tf.identity(inputs, name="residual")
 
         if self.downsample is not None:
-            residual = self.downsample(residual)
+            residual = self.downsample(residual, training=training)
 
         x = self.conv1(inputs)
         x = self.bn1(x, training=training)