Skip to content

GitLab

Commit c0c58423 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 305806848
parent d10d0027
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
official/vision/image_classification/classifier_trainer.py
View file @ c0c58423
...@@ -233,7 +233,6 @@ def initialize(params: base_configs.ExperimentConfig, ...@@ -233,7 +233,6 @@ def initialize(params: base_configs.ExperimentConfig,
dataset_builder: dataset_factory.DatasetBuilder): dataset_builder: dataset_factory.DatasetBuilder):
"""Initializes backend related initializations.""" """Initializes backend related initializations."""
keras_utils.set_session_config( keras_utils.set_session_config(
enable_eager=params.runtime.run_eagerly,
enable_xla=params.runtime.enable_xla) enable_xla=params.runtime.enable_xla)
if params.runtime.gpu_threads_enabled: if params.runtime.gpu_threads_enabled:
keras_utils.set_gpu_thread_mode_and_count( keras_utils.set_gpu_thread_mode_and_count(
......
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