From d51d90a68ffe53fedf7c880fed37499bcb4c992f Mon Sep 17 00:00:00 2001 From: Rino Lee Date: Fri, 10 Jun 2022 05:20:39 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 454141145 --- official/projects/qat/vision/configs/image_classification.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/official/projects/qat/vision/configs/image_classification.py b/official/projects/qat/vision/configs/image_classification.py index 08e01cb65..b3f035697 100644 --- a/official/projects/qat/vision/configs/image_classification.py +++ b/official/projects/qat/vision/configs/image_classification.py @@ -35,6 +35,8 @@ def image_classification_imagenet() -> cfg.ExperimentConfig: task = ImageClassificationTask.from_args( quantization=common.Quantization(), **config.task.as_dict()) config.task = task + runtime = cfg.RuntimeConfig(enable_xla=False) + config.runtime = runtime return config -- GitLab