怎样将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重写一遍哈。
032020-11-17
相似问题