Disable profiling on classifier trainer.

PiperOrigin-RevId: 366883662

Disable profiling on classifier trainer.
PiperOrigin-RevId: 366883662
ccd05aca · Allen Wang · A. Unique TensorFlower · 4c514dc6 · ccd05aca
Commit ccd05aca authored Apr 05, 2021 by Allen Wang Committed by A. Unique TensorFlower Apr 05, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

official/vision/image_classification/callbacks.py official/vision/image_classification/callbacks.py +2 -1

No files found.
--- a/official/vision/image_classification/callbacks.py
+++ b/official/vision/image_classification/callbacks.py
@@ -59,7 +59,8 @@ def get_callbacks(
            log_dir=model_dir,
            track_lr=track_lr,
            initial_step=initial_step,
-            write_images=write_model_weights))
+            write_images=write_model_weights,
+            profile_batch=0))
  if time_history:
    callbacks.append(
        keras_utils.TimeHistory(