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

num_replicas now propagates.

Browse Source
This commit is contained in:
jpodivin 2020-12-20 16:35:46 +00:00
parent 50216cbe96
commit 4bf812c99a
1 changed files with 1 additions and 1 deletions

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

@ -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."""