experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request from jpodivin:reportarg

Browse Source 
PiperOrigin-RevId: 348573959
Change-Id: I20cee2bea44676eb9cf6a17a7a1dab7c8c42889f
This commit is contained in:
TensorFlower Gardener 2020-12-21 21:07:25 -08:00
commit 4967bdb59e
1 changed files with 2 additions and 2 deletions

4
tensorflow/python/eager/benchmarks/resnet50/resnet50_test.py
View File

@ -266,8 +266,8 @@ class ResNet50Benchmarks(tf.test.Benchmark):
def _report(self, label, start, num_iters, device, batch_size, data_format, def _report(self, label, start, num_iters, device, batch_size, data_format,
num_replicas=1): num_replicas=1):
resnet50_test_util.report(self, label, start, num_iters, device, resnet50_test_util.report(self, label, start, num_iters, device, batch_size,
batch_size, data_format, num_replicas=1) data_format, num_replicas)
def _train_batch_sizes(self): def _train_batch_sizes(self):
"""Choose batch sizes based on GPU capability.""" """Choose batch sizes based on GPU capability."""