yetrun's picture
ver3: 将源码迁入 src/deep_learning 包,重塑训练流水线,规范 data/model 契约
07cb7d3
Raw History Blame Contribute Delete
210 Bytes
"""Keras 相关工具模块
提供 Keras 配置相关的功能。
"""
import keras
def enable_mixed_precision():
"""开启混合精度训练/推理"""
keras.config.set_dtype_policy("mixed_float16")