Skip to content

GitLab

Commit 47163394 authored by Haoyu Zhang's avatar Haoyu Zhang Committed by Toby Boyd
Browse files

Fix Keras 1 GPU fp16 graph tests (#6545) 

Forgot to add the `dtype='fp16'` flag in previous CL...
parent 82b928fe
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/resnet/keras/keras_imagenet_benchmark.py
View file @ 47163394
...@@ -291,6 +291,7 @@ class Resnet50KerasBenchmarkBase(keras_benchmark.KerasBenchmark): ...@@ -291,6 +291,7 @@ class Resnet50KerasBenchmarkBase(keras_benchmark.KerasBenchmark):
self._setup() self._setup()
FLAGS.num_gpus = 1 FLAGS.num_gpus = 1
FLAGS.dtype = 'fp16'
FLAGS.enable_eager = False FLAGS.enable_eager = False
FLAGS.distribution_strategy = 'default' FLAGS.distribution_strategy = 'default'
FLAGS.model_dir = self._get_model_dir('benchmark_graph_1_gpu_fp16') FLAGS.model_dir = self._get_model_dir('benchmark_graph_1_gpu_fp16')
...@@ -302,6 +303,7 @@ class Resnet50KerasBenchmarkBase(keras_benchmark.KerasBenchmark): ...@@ -302,6 +303,7 @@ class Resnet50KerasBenchmarkBase(keras_benchmark.KerasBenchmark):
self._setup() self._setup()
FLAGS.num_gpus = 1 FLAGS.num_gpus = 1
FLAGS.dtype = 'fp16'
FLAGS.enable_eager = False FLAGS.enable_eager = False
FLAGS.enable_xla = True FLAGS.enable_xla = True
FLAGS.distribution_strategy = 'default' FLAGS.distribution_strategy = 'default'
......
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