From 4bf812c99a11da9721bc935a022d840fd4e70ccd Mon Sep 17 00:00:00 2001
From: jpodivin <jpodivin@users.noreply.github.com>
Date: Sun, 20 Dec 2020 16:35:46 +0000
Subject: [PATCH] num_replicas now propagates.
---
tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py b/tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py
index 573c8bc2e10..8f8e55f98c6 100644
--- a/tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py
+++ b/tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py
@@ -267,7 +267,7 @@ class ResNet50Benchmarks(tf.test.Benchmark):
def _report(self, label, start, num_iters, device, batch_size, data_format,
num_replicas=1):
resnet50_test_util.report(self, label, start, num_iters, device,
- batch_size, data_format, num_replicas=1)
+ batch_size, data_format, num_replicas)
def _train_batch_sizes(self):
"""Choose batch sizes based on GPU capability."""