From 652ae13acd97cb971f974d248000de0901e75edc Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 13 Sep 2021 10:55:14 -0700 Subject: [PATCH] Tensorboard callback by default should disable profiling. Whether perform profiling should be at user's discretion. PiperOrigin-RevId: 396394295 --- .../golden/v2/tensorflow.keras.callbacks.-tensor-board.pbtxt | 2 +- keras/callbacks.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/keras/api/golden/v2/tensorflow.keras.callbacks.-tensor-board.pbtxt b/keras/api/golden/v2/tensorflow.keras.callbacks.-tensor-board.pbtxt index efe8d0855..afde757a3 100644 --- a/keras/api/golden/v2/tensorflow.keras.callbacks.-tensor-board.pbtxt +++ b/keras/api/golden/v2/tensorflow.keras.callbacks.-tensor-board.pbtxt @@ -6,7 +6,7 @@ tf_class { is_instance: "" member_method { name: "__init__" - argspec: "args=[\'self\', \'log_dir\', \'histogram_freq\', \'write_graph\', \'write_images\', \'write_steps_per_second\', \'update_freq\', \'profile_batch\', \'embeddings_freq\', \'embeddings_metadata\'], varargs=None, keywords=kwargs, defaults=[\'logs\', \'0\', \'True\', \'False\', \'False\', \'epoch\', \'2\', \'0\', \'None\'], " + argspec: "args=[\'self\', \'log_dir\', \'histogram_freq\', \'write_graph\', \'write_images\', \'write_steps_per_second\', \'update_freq\', \'profile_batch\', \'embeddings_freq\', \'embeddings_metadata\'], varargs=None, keywords=kwargs, defaults=[\'logs\', \'0\', \'True\', \'False\', \'False\', \'epoch\', \'0\', \'0\', \'None\'], " } member_method { name: "on_batch_begin" diff --git a/keras/callbacks.py b/keras/callbacks.py index c52b7f529..0676ce3e5 100644 --- a/keras/callbacks.py +++ b/keras/callbacks.py @@ -2073,8 +2073,7 @@ class TensorBoard(Callback, version_utils.TensorBoardVersionSelector): profile_batch: Profile the batch(es) to sample compute characteristics. profile_batch must be a non-negative integer or a tuple of integers. A pair of positive integers signify a range of batches to profile. - By default, it will profile the second batch. Set profile_batch=0 - to disable profiling. + By default, profiling is disabled. embeddings_freq: frequency (in epochs) at which embedding layers will be visualized. If set to 0, embeddings won't be visualized. embeddings_metadata: Dictionary which maps embedding layer names to the @@ -2159,7 +2158,7 @@ class TensorBoard(Callback, version_utils.TensorBoardVersionSelector): write_images=False, write_steps_per_second=False, update_freq='epoch', - profile_batch=2, + profile_batch=0, embeddings_freq=0, embeddings_metadata=None, **kwargs): -- GitLab