Skip to content

GitLab

Unverified Commit 078575a1 authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

added data_dir. (#6205)

parent 68b724ce
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
official/resnet/keras/keras_imagenet_benchmark.py
View file @ 078575a1
...@@ -70,9 +70,11 @@ class Resnet50KerasAccuracy(keras_benchmark.KerasBenchmark): ...@@ -70,9 +70,11 @@ class Resnet50KerasAccuracy(keras_benchmark.KerasBenchmark):
"""Restricts CPU memory allocation.""" """Restricts CPU memory allocation."""
self._setup() self._setup()
FLAGS.num_gpus = 8 FLAGS.num_gpus = 8
FLAGS.enable_eager = True FLAGS.data_dir = DATA_DIR
FLAGS.model_dir = self._get_model_dir('benchmark_8_gpu_bfc_allocator') FLAGS.model_dir = self._get_model_dir('benchmark_8_gpu_bfc_allocator')
FLAGS.dtype = 'fp32'
FLAGS.batch_size = 128 * 8 # 8 GPUs FLAGS.batch_size = 128 * 8 # 8 GPUs
FLAGS.enable_eager = True
# Limits CPU memory to work around memory spikes in eager mode. # Limits CPU memory to work around memory spikes in eager mode.
# TODO(yuefengz): get rid of this test once we fix the memory issue. # TODO(yuefengz): get rid of this test once we fix the memory issue.
os.environ['TF_CPU_ALLOCATOR_USE_BFC'] = 'true' os.environ['TF_CPU_ALLOCATOR_USE_BFC'] = 'true'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment