一行代码迁移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

一行代码迁移TensorFlow 1.x到TensorFlow 2.0相关推荐

【图像分类案例】(2) DenseNet 天气图片四分类（权重迁移学习），附Tensorflow完整代码
各位同学好,今天和大家分享一下使用 Tensorflow 构建 DenseNet 卷积神经网络模型,并使用预训练模型的权重,完成对四种天气图片的分类. 完整代码在我的 Gitee 中,有需要的自取: ...
pytorch 获取模型参数_剑指TensorFlow，PyTorch Hub官方模型库一行代码复现主流模型...
选自PyTorch 机器之心编译参与:思源.一鸣经典预训练模型.新型前沿研究模型是不是比较难调用?PyTorch 团队今天发布了模型调用神器 PyTorch Hub,只需一行代码,BERT.GPT ...
模型秒变API只需一行代码，支持TensorFlow等框架
点击我爱计算机视觉标星,更快获取CVML新技术本文转自机器之心参与:一鸣.杜伟还在为机器学习模型打包成 API 发愁?这个工具能让你一行代码直接打包. 专注于机器学习应用的人们知道,从训练好的模 ...
vscode代码提示只显示一行_模型秒变API只需一行代码，支持TensorFlow等框架
选自GitHub 机器之心编译参与:一鸣.杜伟还在为机器学习模型打包成 API 发愁?这个工具能让你一行代码直接打包. 专注于机器学习应用的人们知道,从训练好的模型到实际的工业生产工具还有一定的距离 ...
神奇的Python-实现曼德布洛特(Mandelbrot)集合(一行代码，matplotlib numpy，tensorflow)分别实现
神奇的Python-实现曼德布洛特(Mandelbrot)集合(一行代码,matplotlib numpy,tensorflow)分别实现 Mandelbrot图实际上是由Mandelbrot集合构成 ...
【CVPR2020 Oral】只需一行代码就可提升迁移性能
关注上方"深度学习技术前沿",选择"星标公众号", 精选文章,第一时间送达! 作者:Hassassin 知乎链接:https://zhuanlan.zhihu. ...
CVPR 2020 Oral：一行代码提升迁移性能，中科院计算所研究生一作
贾浩楠发自凹非寺量子位报道 | 公众号 QbitAI 在迁移学习任务中,一直存在这样的问题: 目标域由于没有标签,常常导致分界面附近混淆较多的数据. 中科院计算所的在读研究生崔书豪等,提出了一 ...
【神经网络】(11) 轻量化网络MobileNetV1代码复现、解析，附Tensorflow完整代码
各位同学好,今天和大家分享一下如何使用 Tensorflow 复现轻量化神经网络模型 MobileNetV1.为了能将神经网络模型用于移动端(手机)和终端(安防监控.无人驾驶)的实时计算,通常这些设备 ...
Spring Cloud Alibaba迁移指南2：一行代码从Hystrix迁移到Sentinel
作者:洛夜,校对:周立在本博客首发,欢迎转载. 前段时间,Netflix宣布Hystrix进入维护模式,详见Hystrix停止开发,我们该何去何从?,而Spring Cloud亦宣布Spring C ...
Spring Cloud Alibaba迁移指南（一）：一行代码从 Hystrix 迁移到 Sentinel
自 Spring Cloud 官方宣布 Spring Cloud Netflix 进入维护状态后,我们开始制作<Spring Cloud Alibaba迁移指南>系列文章,向开发者提供更多 ...

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

警告

一行代码迁移TensorFlow 1.x到TensorFlow 2.0相关推荐

最新文章

热门文章