Skip to content

Commit 04b94a5

Browse files
Allen Wangaman2930
authored andcommitted
Disable profiling on classifier trainer.
PiperOrigin-RevId: 366883662
1 parent 912b54b commit 04b94a5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

official/vision/image_classification/callbacks.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,8 @@ def get_callbacks(
5959
log_dir=model_dir,
6060
track_lr=track_lr,
6161
initial_step=initial_step,
62-
write_images=write_model_weights))
62+
write_images=write_model_weights,
63+
profile_batch=0))
6364
if time_history:
6465
callbacks.append(
6566
keras_utils.TimeHistory(

0 commit comments

Comments
 (0)