升级到tensorflow2.0

2023-05-16

目录

  • 从tensorflow1.x升级到2.x
    • 方案一:依然使用tf 1.x的脚本
    • 方案二:升级项目代码到2.x

tensorflow2.0推出以后,全面拥抱keras,简化了API接口,使得tensorflow在使用上宜人了很多。

从tensorflow1.x升级到2.x

方案一:依然使用tf 1.x的脚本

对于原来的1.x系列代码,当后台的tensorflow支持升级到2.x以后,大多数代码并不需要重新改写,只需要将加载tensorflow的代码修改为

import tensorflow.compat.v1 as tf

tf.disable_v2_behavior()

就可以了。

方案二:升级项目代码到2.x

但是这样做并没有真正发挥2.x新的API带来的新功能。为了将1.x的项目代码真正迁移成2.x风格的,tensorflow项目组发布了tf_upgrade_v2脚本实现这个功能。该脚本的使用方法如下:

tf_upgrade_v2 \

  --intree my_project/ \

  --outtree my_project_v2/ \

  --reportfile report.txt

其他参数可以查看tf_upgrade_v2脚本的帮助文档:tf_upgrade_v2 -h

但是有些API的符号并不能仅仅通过字符串替换简单的得到。虽然大多数1.x版本tf函数的调用都可以通过修改调用函数为tf.compat.v1.func_name()的方式得到,但是也有部分函数在2.x版本中被弃用了,例如tf.flags和tf.contrib, 另外还有一些函数可能需要调用一些其他的哭,例如absl.flags,或者将对应的函数调整到了tensorflow.addons模块中。

鉴于以上情况,官方推荐的升级方式为:
1)单元测试
2)安装tensorflow 1.14, 在tf1.14版本中实际上已经包括了tf 2.0版本的代码,封装在tf.compat.v2模块中。
3)用tf 1.14版本的代码进行测试,这一步非常关键,确保所有的代码在1.14版本的测试用全部通过。
4)用tf_upgrade_v2升级项目代码
5)在用1.14版本测试升级后的代码。
6)检查upgrade报告中的warning和error
7)安装tf2.0
8)测试代码,代码前面加入tf.disable_v2_behavior()函数,再运行测试
9)注释掉测试代码中的 ,用2.x的API运行。

如果遇到什么问题可以查看官方代码迁移指南:https://tensorflow.google.cn/guide/migrate

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

升级到tensorflow2.0 的相关文章

随机推荐

  • HIDL转AIDL编译报错:ISecureElement does not have VINTF level stablity, but interface requires it.

    在m vendor sprd hardware secure element update api生成stable接口时 xff0c 总是报错如下 找了一圈找不到解决方法 xff0c 最后误打误撞地解决了 xff0c 供大家参考 HIDL转
  • Mac上的远程连接工具Royal TSX,比FinalShell更值得被推荐

    安装Royal TSX xff1a https blog csdn net Darling qi article details 120289137 使用Royal TSX xff1a https blog csdn net Bluffin
  • libjvm.so: ELF file OS ABI invalid

    Error dl failure on line 893 Error failed 某目录 jdk jre lib amd64 server libjvm so because 某目录 jdk jre lib amd64 server li
  • win11启用旧右键菜单(不折叠)的方案,亲测有效

    我电脑的winodws11版本如下 在以上win版本下 xff0c 测试网上说的以下几个办法都不行 1 权利与暗访时在终端运行 reg exe delete HKCU Software Classes CLSID 86ca1aa0 34aa
  • Android中的动画总结

    Android 动画 三种总结 xff1a 属性动画 xff1a 动态的改变属性产生动画效果 xff1b 改变动画的属性 xff0c 两个重要的类 xff1a 1 ValueAnimator 类是先改变值 xff0c 然后 手动赋值 给对象
  • ubuntu自动登录tty终端的最简方法

    背景 在嵌入式系统经常需要自动登录tty xff0c 以实现业务程序开机启动的效果 网上有篇文章ubuntu自动登录tty1 shell text 配置转发挺多 xff0c 但我弄明白原理后 xff0c 觉得可以进一步简化 xff0c 经测
  • git为指定项目设置用户名密码

    我们如果没有为项目设置用户名密码 xff0c 那么每次提交都会有提示账号和密码输入 xff1a 1 找到你项目的半隐藏文件 git文件夹 xff0c 通常只要git init 就会生成这样一个文件夹 xff0c 现在双击进入文件夹 xff1
  • JAVA 访问windows共享文件夹

    一 使用技术 JCIFS框架对Windows共享文件夹进行读写 xff0c 使用了SMB通信协议 xff0c 它为局域网内不同的计算机之间提供文件和打印机等资源共享服务 二 共享文件夹设置 测试共享文件夹机器windows版本为win10家
  • 前端代码调试:Webstorm调试js

    前言 目前前端开发 JavaScript的debug一般都是用chrome和firefox的开发者工具进行调试 xff0c 浏览器工具使用不方便 xff0c webstorm支持了在代码上打断点 xff0c 在编辑器里debug js代码
  • CV:基本概念和工具

    文章目录 计算机视觉像素 xff0c 颜色 xff0c 通道 xff0c 图像和颜色空间的概念图像文件后缀有损压缩和无损压缩 Python的计算机视觉库PIL xff1a python 图像处理类库载入PIL库基本操作命令 openCVop
  • OCR技术概览

    OCR技术概览 OCR Optical Character Recognition 光学字符识别技术主要分为手写体识别和印刷体识别两类 印刷体识别比手写体识别要简单 因为印刷体更规范 字体来自于计算机字库 尽管印刷过程中可能会发生不清晰 粘
  • CV:图像色彩空间及色彩处理

    文章目录 基本概念RGB空间HSV空间HSL空间 色彩变换灰度变换色彩反向 调整像素区间增强对比度直方图均衡化 图像平滑 减少噪声图像平均高斯滤波 图像梯度sobel算子 scharr算子prewitt算子Laplacian 算子 参考及更
  • OpenCV角点检测: Harris算子, ShiTomasi算子

    角点检测 角点的特征检测与匹配是Computer Vision 应用总重要的一部分 xff0c 这需要寻找图像之间的特征建立对应关系 点 xff0c 也就是图像中的特殊位置 xff0c 是很常用的一类特征 xff0c 点的局部特征也可以叫做
  • 集成学习:让算法和算法赛跑

    文章目录 集成学习的基本概念构建弱分类器 xff1a 决策树自助采样法bootstrappingbootstrapping的核心思想bootstrapping与permutation的区别 baggingboosting为何bagging会
  • backbone模型:FCN、SRN、STN

    文章目录 FCN网络CNN图像分割模型结构FCN github资源FCN的优缺点 SRN 网络什么是空间规整 spatial regularization xff09 SRN网络github资源 STN网络空间变换器localisation
  • docker部署机器学习/深度学习模型的容器化方案

    文章目录 什么是dockerdocker的优点 docker image镜像Dockerfile 文件Dockerfile配置例子 创建docker镜像 docker container 容器模型部署参考和更多阅读 docker部署机器学习
  • RNN模型训练经验总结

    文章目录 RNN模型训练经验总结数据准备 look at your data 小步试错 搭建模型设置端到端的训练评估框架 forward propagation设置激活函数dropout back propagation设置学习率 lear
  • 算法的时间复杂度和空间复杂度

    如何评价算法的性能 定义 一个算法中的语句执行次数称为 语句频度 或 时间频度 约定 检验算法的效率 xff0c 主要考虑 最坏时间复杂度 和 平均时间复杂度 一般不特别说明 xff0c 讨论的时间复杂度均是最坏情况下的时间复杂度 时间复杂
  • 【LeetCode】LCS最长公共子序列

    最长公共子序列 题目描述思路分析递归结构算法实现输出最长子序列算法实现 题目描述 思路分析 设A 61 a0 xff0c a1 xff0c xff0c am xff0c B 61 b0 xff0c b1 xff0c xff0c bn xff
  • 升级到tensorflow2.0

    目录 从tensorflow1 x升级到2 x方案一 xff1a 依然使用tf 1 x的脚本方案二 xff1a 升级项目代码到2 x tensorflow2 0推出以后 xff0c 全面拥抱keras xff0c 简化了API接口 xff0