当前位置: 首页 > news >正文

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

\"image\"

注意:通过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

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

\"image\"

添加关键字:

\"image\"

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

\"image\"

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

\"\"

警告

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

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

参考链接:
https://medium.com/tensorflow/upgrading-your-code-to-tensorflow-2-0-f72c3a4d83b5

更多内容,请关注AI前线

\"image\"

相关文章:

  • Oracle 12c 数据库中scott用户不存在的解决方法
  • redis使用场景和java测试案例
  • 数据库迁移
  • 无监督学习新突破!华为美研所提出自动编码变换网络AET
  • YII AssetManager分析 最小化脚本文件
  • 迅速搭建简易静态服务器
  • 在CMD窗口中使用javac和java命令进行编译和执行带有包名的具有继承关系的类
  • 4.56MariaDB密码重置4.57MariaDB慢查询日志
  • 第一次作业(2)
  • 基于遗传算法的优化问题求解
  • LSMTree - SStable 初体验
  • C++回声服务器_9-epoll边缘触发模式版本服务器
  • 242. Valid Anagram(C++)
  • 冒泡排序及回调函数的使用
  • HTML5基础(四)
  • hexo+github搭建个人博客
  • CSS 专业技巧
  • ES6之路之模块详解
  • javascript从右向左截取指定位数字符的3种方法
  • JavaScript工作原理(五):深入了解WebSockets,HTTP/2和SSE,以及如何选择
  • Java知识点总结(JavaIO-打印流)
  • jQuery(一)
  • Median of Two Sorted Arrays
  • Meteor的表单提交:Form
  • puppeteer stop redirect 的正确姿势及 net::ERR_FAILED 的解决
  • Spring-boot 启动时碰到的错误
  • VuePress 静态网站生成
  • vue-router的history模式发布配置
  • 表单中readonly的input等标签,禁止光标进入(focus)的几种方式
  • 关于 Linux 进程的 UID、EUID、GID 和 EGID
  • 基于MaxCompute打造轻盈的人人车移动端数据平台
  • 聊聊hikari连接池的leakDetectionThreshold
  • 前端每日实战:70# 视频演示如何用纯 CSS 创作一只徘徊的果冻怪兽
  • 如何设计一个比特币钱包服务
  • ​Linux·i2c驱动架构​
  • ​猴子吃桃问题:每天都吃了前一天剩下的一半多一个。
  • #stm32整理(一)flash读写
  • (02)vite环境变量配置
  • (1)Android开发优化---------UI优化
  • (C语言)字符分类函数
  • (DFS + 剪枝)【洛谷P1731】 [NOI1999] 生日蛋糕
  • (笔记)Kotlin——Android封装ViewBinding之二 优化
  • (草履虫都可以看懂的)PyQt子窗口向主窗口传递参数,主窗口接收子窗口信号、参数。
  • (附源码)计算机毕业设计ssm本地美食推荐平台
  • (剑指Offer)面试题41:和为s的连续正数序列
  • (蓝桥杯每日一题)love
  • (切换多语言)vantUI+vue-i18n进行国际化配置及新增没有的语言包
  • .bashrc在哪里,alias妙用
  • .NET 除了用 Task 之外,如何自己写一个可以 await 的对象?
  • .net 验证控件和javaScript的冲突问题
  • .NET/C# 使用 ConditionalWeakTable 附加字段(CLR 版本的附加属性,也可用用来当作弱引用字典 WeakDictionary)
  • /bin/bash^M: bad interpreter: No such file ordirectory
  • /var/log/cvslog 太大
  • :如何用SQL脚本保存存储过程返回的结果集
  • [2021ICPC济南 L] Strange Series (Bell 数 多项式exp)