We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 912b54b commit 04b94a5Copy full SHA for 04b94a5
official/vision/image_classification/callbacks.py
@@ -59,7 +59,8 @@ def get_callbacks(
59
log_dir=model_dir,
60
track_lr=track_lr,
61
initial_step=initial_step,
62
- write_images=write_model_weights))
+ write_images=write_model_weights,
63
+ profile_batch=0))
64
if time_history:
65
callbacks.append(
66
keras_utils.TimeHistory(
0 commit comments