From 6e412ad7725ec68c2e771149af953160420e9c4e Mon Sep 17 00:00:00 2001 From: Nihal Harish Date: Wed, 20 Nov 2019 15:46:01 -0800 Subject: [PATCH] Fix renaming typo - inputs. was renamed to inpusmd --- examples/tensorflow/scripts/train_imagenet_resnet_hvd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/tensorflow/scripts/train_imagenet_resnet_hvd.py b/examples/tensorflow/scripts/train_imagenet_resnet_hvd.py index 0184b3873..f1a1ae098 100644 --- a/examples/tensorflow/scripts/train_imagenet_resnet_hvd.py +++ b/examples/tensorflow/scripts/train_imagenet_resnet_hvd.py @@ -176,7 +176,7 @@ def batch_norm(self, inputs, **kwargs): ) def spatial_average2d(self, inputs): - shape = inpusmd.get_shape().as_list() + shape = inputs.get_shape().as_list() if self.data_format == "channels_last": n, h, w, c = shape else: