提交 652ae13a 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Tensorboard callback by default should disable profiling.

Whether perform profiling should be at user's discretion.

PiperOrigin-RevId: 396394295
上级 6859a01c
......@@ -6,7 +6,7 @@ tf_class {
is_instance: "<type \'object\'>"
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"
......
......@@ -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):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册