diff options
author | 2017-06-27 10:30:08 -0700 | |
---|---|---|
committer | 2017-06-27 10:40:26 -0700 | |
commit | 3ded50e4827ba8e75d6200ee8102662d3f9caf86 (patch) | |
tree | 3c9d4f061be63e625df08d2465e822c6e4c9e7ce /tensorflow | |
parent | 4114aaebbe68bf37e73204faf7e3392586c3c1fb (diff) |
Shrink the test size to avoid OOM error on old GPUs.
PiperOrigin-RevId: 160292834
Diffstat (limited to 'tensorflow')
-rw-r--r-- | tensorflow/python/ops/conv2d_benchmark.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/ops/conv2d_benchmark.py b/tensorflow/python/ops/conv2d_benchmark.py index c6352b2a98..6992fa57ea 100644 --- a/tensorflow/python/ops/conv2d_benchmark.py +++ b/tensorflow/python/ops/conv2d_benchmark.py @@ -120,8 +120,8 @@ class Conv2DBenchmark(test.Benchmark): def benchmark_conv2d(self): print("conv2d benchmark:") - h = 1000 - w = 1000 + h = 500 + w = 500 fh = 3 fw = 3 input_shapes = [] |