keras混合精度训练代码

先上配置tf 1.14 keras 2.2.4 GeForce RTX 2080 Ti *1 cuda 10.1 python 3.6 很是很是简单几行代码python import os os.environ['TF_AUTO_MIXED_PRECISION_GRAPH_REWRITE_IGNORE_PERFORMANCE'] = '1' import tensorflow as tf opt=
相关文章
相关标签/搜索