Internal change

PiperOrigin-RevId: 454141145

Internal change
PiperOrigin-RevId: 454141145
eb4452c5 · Rino Lee · A. Unique TensorFlower · 2bc29a6f · eb4452c5
Commit eb4452c5 authored Jun 10, 2022 by Rino Lee Committed by A. Unique TensorFlower Jun 10, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/projects/qat/vision/configs/image_classification.py official/projects/qat/vision/configs/image_classification.py +2 -0

No files found.
--- 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