写点什么

一行代码迁移 TensorFlow 1.x 到 TensorFlow 2.0

  • 2019-03-01
  • 本文字数:1021 字

    阅读完需:约 3 分钟

一行代码迁移TensorFlow 1.x到TensorFlow 2.0

TensorFlow 2.0 将会包含许多 API 改变,例如改变参数顺序、重命名、修改默认值等。手动迁移代码比较繁琐,还有出错的风险。为了让你的老代码能够向 TensorFlow 2.0 无缝迁移,TensorFlow 开发组推出了 tf_upgrade_v2 功能。


当你使用 pip 安装 TensorFlow 2.0 时,tf_upgrade_v2 就会自动安装,它可以帮你更快速地将已有的 TensorFlow 1.13 脚本迁移到 TensorFlow 2.0。


我们尝试尽可能多地自动执行升级任务:但是,仍然有一些句法和风格改变是 tf_upgrade_v2 无法自动迁移的,这些变更无法用简单的字符串替换来实现。为了确保你的代码可以被 TensorFlow 2.0 支持,升级脚本中包含了一个 compat.v1 模块。这个模块会将 tf.foo 形式的调用替换为 tf.compat.v1.foo。我们推荐的做法是,尽快人工校对这些替换,并将它们迁移到 tf.*命名空间下的新 API 而不是使用 tf.compat.v1.*命名空间。


另外,由于一些模块被废弃(例如 tf.flags 和 tf.contrib),TensorFlow 2.0 会包含一些无法通过切换到 compat.v1 解决的变更。升级这些代码需要用到额外的库(如 absl.flags)或者将代码切换到 tensorflow/addons 包下。


如果你想要把模型从 TensorFlow 1.12 迁移到 TensorFlow 2.0,首先要安装 tf-nightly-2.0-preview 或 tf-nightly-gpu-2.0-preview。


然后就可以通过运行一个单独的 Python 脚本直接升级:


tf_upgrade_v2 — infile foo.py — outfile foo-upgraded.py
复制代码



注意:通过 pip install 安装 TensorFlow 1.13 及后续版本时会自动安装 tf_upgrade_v2。


也可以在文件夹目录中执行升级脚本:


# upgrade the .py files and copy all the other files to the outtreetf_upgrade_v2 — intree foo/ — outtree foo-upgraded/# just upgrade the .py filestf_upgrade_v2 — intree foo/ — outtree foo-upgraded/ — copyotherfiles False
复制代码


脚本会列出到底做了哪些改动,例如参数重命名:



添加关键字:



以及有哪些地方推荐进行人工检查:



所有的信息都会被包含在当前文件夹的 report.txt 中。升级完成后,你可以运行模型来检查你的输出是否和 1.13 版本的输出相似。


警告

  • 在运行脚本之前,不要手动修改代码。尤其是那些会变更参数顺序的函数,例如 tf.argmax 或者 tf.batch_to_space,这会导致脚本错误地映射参数关键词。

  • 脚本本身不会进行参数重排序,只会为发生了参数重排的函数添加参数关键字


参考链接:


https://medium.com/tensorflow/upgrading-your-code-to-tensorflow-2-0-f72c3a4d83b5


更多内容,请关注 AI 前线



2019-03-01 13:0011338
用户头像
蔡芳芳 InfoQ 总编辑

发布了 813 篇内容, 共 590.4 次阅读, 收获喜欢 2808 次。

关注

评论

发布
暂无评论
发现更多内容

架构师训练营大作业二

Hanson

架构师训练营第一周

子青

LeetCode题解:94. 二叉树的中序遍历,递归,JavaScript,详细注释

Lee Chen

大前端 LeetCode

第一周命题作业

崔方剑

极客大学架构师训练营

week1-UML图

张兵

极客大学架构师训练营

第1周 架构方法 浮皮潦草之总结

Pyr0man1ac

「架构师训练营第 1 期」-食堂卡管理系统

睡不着摇一摇

极客大学架构师训练营

区块链将掀开人类的伟大时代

CECBC

区块链 智能合约 价值物联网

第一周作业一:食堂就餐卡系统设计

登顶计划

极客大学架构师训练营

go runtime debug 小技巧

Gopher指北

debug 后端 runtime Go 语言

我们需要软件工艺

Bruce Talk

敏捷 随笔 Agile

架构师训练营第一周作业 (就餐卡UML图)

springH₂O

极客时间架构师培训1期-第1周作业

Kaven

架构师训练营第一期第一周学习总结

朱磊

极客大学架构师训练营

架构师训练营第 1 期-第一周周总结

郑凯元

极客大学架构师训练营

极客时间架构师培训1期-第1周总结

Kaven

架构师训练营第 1 期-第一周学习总结

Todd-Lee

极客大学架构师训练营

第一周学习总结

崔方剑

极客大学架构师训练营

食堂就餐卡系统设计

……

电商管理系统之交易子系统设计(一)

长沙造纸农

系统设计 产品经理 系统架构 订单管理 电商平台

项目滞后,如何让自己的技术快速成长

郎哲158

个人成长 舒适区 熟练工

训练营第一周作业 2

仲夏

训练营第一周作业1

仲夏

架构师训练营第一期第一周命题作业

朱磊

极客大学架构师训练营

极客大学--架构师训练营1期-第一周总结(vaik)

行之

架构师训练营第一周作业

木头发芽

中国法定数字货币发展新机遇

CECBC

数字货币 数字经济

架构师训练营第 1 期 - 作业提交

Todd-Lee

极客大学架构师训练营

用简单而又专业的角度为大家揭秘区块链和比特币

CECBC

比特币 区块链 数字货币

Spring Cloud 微服务实践 (3) - 服务间的调用

xiaoboey

Spring Cloud 熔断 服务调用 Feign

RxSwift和RxCocoa入门

teoking

ios swift

一行代码迁移TensorFlow 1.x到TensorFlow 2.0_AI&大模型_TensorFlow团队_InfoQ精选文章