怎样将TF1.4代码改成TF2.0以上

来源:5-4 keras_to_estimator

OCEANDREAM

2020-11-13

老师,您好,想请问一下,怎样把tensorflow 1.4的代码改成tensorflow 2.0的,不知道需要修改哪些内容。代码是从TF 0.12 改到TF 1.4, 现在想把代码改成TF2.0 。tensorflow 1.4的版本是多GPU版本的,使用的还是静态图。

写回答

1回答

正十七

2020-11-16

迁移的话,大概需要做的就是API都要改,比如compat.v1下的API要改成tf2.0里对应的。同时,sess.run要去掉,静态图可以考虑tf.function等。

可以参考官方给出的文档来做转换: https://www.tensorflow.org/guide/migrate

另外,学完咱们这门课后,同学应该有能力把代码用tf2.0重写一遍哈。

0
3
OCEANDREAM
还有老师,请教您的这个问题,“为什么要转estimator,有什么好处吗”这个看不到您的回复
2020-11-17
共3条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程