diff options
author | Igor Ganichev <iga@google.com> | 2018-08-23 17:18:12 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-23 17:21:24 -0700 |
commit | bdf526389b3dda9f79c450c4806aeb53be81bf58 (patch) | |
tree | bf824edb10e676c1f6c495449831181dcdb6092d /tensorflow/contrib/eager | |
parent | b396ea0ae689fd063f475238289c5e65457a987a (diff) |
Make eager ResNet50 call() method params conformant to convention
PiperOrigin-RevId: 210028452
Diffstat (limited to 'tensorflow/contrib/eager')
-rw-r--r-- | tensorflow/contrib/eager/python/examples/resnet50/resnet50.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/eager/python/examples/resnet50/resnet50.py b/tensorflow/contrib/eager/python/examples/resnet50/resnet50.py index a28bc8a43d..3f70f573b1 100644 --- a/tensorflow/contrib/eager/python/examples/resnet50/resnet50.py +++ b/tensorflow/contrib/eager/python/examples/resnet50/resnet50.py @@ -272,8 +272,8 @@ class ResNet50(tf.keras.Model): else: self.global_pooling = None - def call(self, input_tensor, training): - x = self.conv1(input_tensor) + def call(self, inputs, training=True): + x = self.conv1(inputs) x = self.bn_conv1(x, training=training) x = tf.nn.relu(x) x = self.max_pool(x) |