Spaces:
Sleeping
Sleeping
File size: 210 Bytes
a5fd608 | 1 2 3 4 5 6 7 8 9 10 11 12 | """Keras 相关工具模块
提供 Keras 配置相关的功能。
"""
import keras
def enable_mixed_precision():
"""开启混合精度训练/推理"""
keras.config.set_dtype_policy("mixed_float16")
|