ResNet saving outputs to wrong variable

aidangomez · Ryan Sepassi · commit fb5accee8994 · 2018-04-20T11:44:38.000-07:00
PiperOrigin-RevId: 193558614
diff --git a/tensor2tensor/models/resnet.py b/tensor2tensor/models/resnet.py
@@ -421,7 +421,7 @@ def body(self, features):
         is_training=hp.mode == tf.estimator.ModeKeys.TRAIN)
 
     if hp.use_nchw:
-      inputs = tf.transpose(out, [0, 2, 3, 1])
+      out = tf.transpose(out, [0, 2, 3, 1])
 
     return out
 
