Keras 配置实用工具

[源代码]

version 函数

keras.version()

[源代码]

clear_session 函数

keras.utils.clear_session(free_memory=True)

重置 Keras 生成的所有状态。

Keras 管理一个全局状态，它使用该状态来实现函数式模型构建 API 并为自动生成的层名称生成唯一标识符。

如果您在循环中创建了许多模型，此全局状态会随着时间的推移消耗越来越多的内存，并且您可能需要清除它。调用 clear_session() 会释放全局状态：这有助于避免因旧模型和层而产生的混乱，尤其是在内存有限的情况下。

参数

• free_memory：是否调用 Python 垃圾回收。调用它以确保被删除对象使用的内存立即被释放通常是一个好习惯。但是，它的执行可能需要几秒钟，因此当在短循环中使用 clear_session() 时，您可能希望跳过它。

示例 1：在循环创建模型时调用 clear_session()

for _ in range(100):
  # Without `clear_session()`, each iteration of this loop will
  # slightly increase the size of the global state managed by Keras
  model = keras.Sequential([
      keras.layers.Dense(10) for _ in range(10)])

for _ in range(100):
  # With `clear_session()` called at the beginning,
  # Keras starts with a blank state at each iteration
  # and memory consumption is constant over time.
  keras.backend.clear_session()
  model = keras.Sequential([
      keras.layers.Dense(10) for _ in range(10)])

示例 2：重置层名称生成计数器

>>> layers = [keras.layers.Dense(10) for _ in range(10)]
>>> new_layer = keras.layers.Dense(10)
>>> print(new_layer.name)
dense_10
>>> keras.backend.clear_session()
>>> new_layer = keras.layers.Dense(10)
>>> print(new_layer.name)
dense

[源代码]

enable_traceback_filtering 函数

keras.config.enable_traceback_filtering()

开启堆栈跟踪过滤。

原始 Keras 堆栈跟踪（也称为堆栈跟踪）包含许多内部帧，这可能很难阅读，并且对最终用户没有帮助。默认情况下，Keras 会过滤掉 Keras 引发的大多数异常中的内部帧，以使堆栈跟踪保持简短、可读且关注于对您（您的代码）可操作的内容。

另请参阅 keras.config.disable_traceback_filtering() 和 keras.config.is_traceback_filtering_enabled()。

如果您之前通过 keras.config.disable_traceback_filtering() 禁用了堆栈跟踪过滤，您可以通过 keras.config.enable_traceback_filtering() 重新启用它。

[源代码]

disable_traceback_filtering 函数

keras.config.disable_traceback_filtering()

关闭堆栈跟踪过滤。

原始 Keras 堆栈跟踪（也称为堆栈跟踪）包含许多内部帧，这可能很难阅读，并且对最终用户没有帮助。默认情况下，Keras 会过滤掉 Keras 引发的大多数异常中的内部帧，以使堆栈跟踪保持简短、可读且关注于对您（您的代码）可操作的内容。

另请参阅 keras.config.enable_traceback_filtering() 和 keras.config.is_traceback_filtering_enabled()。

如果您之前通过 keras.config.disable_traceback_filtering() 禁用了堆栈跟踪过滤，您可以通过 keras.config.enable_traceback_filtering() 重新启用它。

[源代码]

is_traceback_filtering_enabled 函数

keras.config.is_traceback_filtering_enabled()

检查是否启用了堆栈跟踪过滤。

原始 Keras 堆栈跟踪（也称为堆栈跟踪）包含许多内部帧，这可能很难阅读，并且对最终用户没有帮助。默认情况下，Keras 会过滤掉 Keras 引发的大多数异常中的内部帧，以使堆栈跟踪保持简短、可读且关注于对您（您的代码）可操作的内容。

另请参阅 keras.config.enable_traceback_filtering() 和 keras.config.disable_traceback_filtering()。

如果您之前通过 keras.config.disable_traceback_filtering() 禁用了堆栈跟踪过滤，您可以通过 keras.config.enable_traceback_filtering() 重新启用它。

返回

布尔值，如果启用了堆栈跟踪过滤，则为 True，否则为 False。

[源代码]

enable_interactive_logging 函数

keras.config.enable_interactive_logging()

开启交互式日志记录。

启用交互式日志记录后，Keras 会通过 stdout 显示日志。这在使用 Keras 的交互式环境（如 shell 或 notebook）时能提供最佳体验。

[源代码]

disable_interactive_logging 函数

keras.config.disable_interactive_logging()

关闭交互式日志记录。

禁用交互式日志记录后，Keras 会将日志发送到 absl.logging。当以非交互方式使用 Keras 时（例如，在服务器上运行训练或推理作业），这是最佳选择。

[源代码]

is_interactive_logging_enabled 函数

keras.config.is_interactive_logging_enabled()

检查是否启用了交互式日志记录。

要在将日志写入 stdout 和 absl.logging 之间切换，您可以使用 keras.config.enable_interactive_logging() 和 keras.config.disable_interactive_logging()。

返回

布尔值，如果启用了交互式日志记录，则为 True，否则为 False。

[源代码]

enable_unsafe_deserialization 函数

keras.config.enable_unsafe_deserialization()

全局禁用安全模式，允许反序列化 lambda。

[源代码]

floatx 函数

keras.config.floatx()

返回默认浮点类型，作为字符串。

例如 'bfloat16'、'float16'、'float32'、'float64'。

返回

字符串，当前默认的浮点类型。

示例

>>> keras.config.floatx()
'float32'

[源代码]

set_floatx 函数

keras.config.set_floatx(value)

设置默认浮点 dtype。

注意：不建议将此设置为 "float16" 进行训练，因为这很可能会导致数值稳定性问题。相反，混合精度利用了 float16 和 float32 的混合。可以通过调用 keras.mixed_precision.set_dtype_policy('mixed_float16') 进行配置。

参数

• value：字符串；'bfloat16'、'float16'、'float32' 或 'float64'。

示例

>>> keras.config.floatx()
'float32'

>>> keras.config.set_floatx('float64')
>>> keras.config.floatx()
'float64'

>>> # Set it back to float32
>>> keras.config.set_floatx('float32')

引发

• ValueError：如果值无效。

[源代码]

image_data_format 函数

keras.config.image_data_format()

返回默认图像数据格式约定。

返回

字符串，可以是 'channels_first' 或 'channels_last'。

示例

>>> keras.config.image_data_format()
'channels_last'

[源代码]

set_image_data_format 函数

keras.config.set_image_data_format(data_format)

设置图像数据格式约定值。

参数

• data_format：字符串。'channels_first' 或 'channels_last'。

示例

>>> keras.config.image_data_format()
'channels_last'

>>> keras.config.set_image_data_format('channels_first')
>>> keras.config.image_data_format()
'channels_first'

>>> # Set it back to `'channels_last'`
>>> keras.config.set_image_data_format('channels_last')

[源代码]

epsilon 函数

keras.config.epsilon()

返回数值表达式中使用的模糊因子值。

返回

一个浮点数。

示例

>>> keras.config.epsilon()
1e-07

[源代码]

set_epsilon 函数

keras.config.set_epsilon(value)

设置数值表达式中使用的模糊因子值。

参数

• value：浮点数。epsilon 的新值。

示例

>>> keras.config.epsilon()
1e-07

>>> keras.config.set_epsilon(1e-5)
>>> keras.config.epsilon()
1e-05

>>> # Set it back to the default value.
>>> keras.config.set_epsilon(1e-7)

[源代码]

backend 函数

keras.config.backend()

用于确定当前后端的可公开访问的方法。

返回

字符串，Keras 当前使用的后端名称。可以是 "tensorflow"、"torch" 或 "jax" 之一。

示例

>>> keras.config.backend()
'tensorflow'