diff --git a/tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py b/tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py index 573c8bc2e10..0a47209cb85 100644 --- a/tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py +++ b/tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py @@ -266,8 +266,8 @@ 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) + resnet50_test_util.report(self, label, start, num_iters, device, batch_size, + data_format, num_replicas) def _train_batch_sizes(self): """Choose batch sizes based on GPU capability."""