使用混合精度 mixed precision 给 Keras 加速
可以使用混合精度 mixed precision 给 Keras 加速,3个操作步骤如下:使用算力在 7.0以上的GPU,比如 NVIDIA的 RTX 3090, 3080等。在建立模型之前,设置 global_policy 为 mixed_float16。policy = tf.keras.mixed_precision.Policy('mixed_float16')tf.keras.mixed_precision.set_global_policy(policy)# 另一种写法是只用下面这