上传者: 38690508
|
上传时间: 2021-05-01 19:31:32
|
文件大小: 47KB
|
文件类型: PDF
下面将升级代码到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工具并不是万能的