将TensorFlow 1.x升级代码到TensorFlow 2.x版本的方法汇总
下面将升级代码到TensorFlow 2.x版本的方法汇总起来,有如下几点。 1.最快速转化的方法 在代码中没有使用contrib模块的情况下,可以在代码最前端加上如下两句,直接可以实现的代码升级。 import tensorflow.compat.v1 as tftf.disable_v2_behavior() 这种方法只是保证代码在TensorFlow 2.x版本上能够运行,并不能发挥TensorFlow的最大性能。 2.使用工具进行转化的方法 在代码中没有使用contrib模块的情况下,用tf_upgrade_v2工具可以快速实现代码升级。当然tf_upgrade_v2工具并不是万能的
暂无评论