yetrun's picture
ver1: 实现深度学习训练框架,支持 Wiki GPT 与诗歌生成双任务
a5fd608
"""Keras 相关工具模块
提供 Keras 配置相关的功能。
"""
import keras
def enable_mixed_precision():
"""开启混合精度训练/推理"""
keras.config.set_dtype_policy("mixed_float16")