环境

ubuntu 18.04
Python 3.8
TensorFlow-gpu 2.3.1
CUDA 11.1
tensorflow-yolov5

报错

WARNING:tensorflow:Inconsistent references when loading the checkpoint into this object graph. Either the Trackable object references in the Python program have changed in an incompatible way, or the checkpoint was generated in an incompatible program.

Two checkpoint references resolved to different objects

训练的时候看不出来问题，可以保存模型。
以上警告发生在model.load_weights时

解决方案

比较像内部bug，直接保存为pb模型则没有问题。保存、预测都可以成功

model.save_weights("../weights/yolo")
# 替换为：
tf.saved_model.save(model, "../weights/yolo")

参考

colab上也有一个case：linkage
有类似报错，把两个function重写为一个解决了

更多内容，欢迎关注我的公众号：YueTan

Debug Tensorflow ：Two checkpoint references resolved to different objects相关推荐

Tensorflow：TF模型文件(checkpoint文件夹下ckpt文件之data、index、meta)保存、模型导入、恢复并fine-tuning之详细攻略
Tensorflow:TF模型文件(checkpoint文件夹下ckpt文件之data.index.meta)保存.模型导入.恢复并fine-tuning之详细攻略目录保存TF训练好的模型 1.T ...
TensorFlow：将ckpt文件固化成pb文件
TensorFlow:将ckpt文件固化成pb文件本文是将yolo3目标检测框架训练出来的ckpt文件固化成pb文件,主要利用了GitHub上的该项目. 为什么要最终生成pb文件呢?简单来说就是直接 ...
TensorFlow 2.0 - Checkpoint 保存变量、TensorBoard 训练可视化
文章目录 1. Checkpoint 保存变量 2. TensorBoard 训练过程可视化学习于:简单粗暴 TensorFlow 2 1. Checkpoint 保存变量 tf.train.Che ...
解决Tensorflow：Could not load dynamic library ‘cudart64_101.dll‘；dynamic library ‘cublas64_10.dll‘；
Introduce Tensorflow: 2.3.0 Cuda: 10.2 VS Code 之前有装pytorch的过程,但是不少文章的模型用的tensrflow..就是就pip install t ...
C 预处理器 —— __DATE__ # __TIME__ # __FILE__ # __LINE__ # __STDC__ (预处理宏的使用 —— 打印debug信息：)
预定义宏: ANSI C 定义了许多宏.在编程中您可以使用这些宏,但是不能直接修改这些预定义的宏. 宏描述 __DATE__ 当前日期,一个以 "MMM DD YYYY&quo ...
当微信小程序遇上TensorFlow：Server端实现补充
在前面一篇文章<当微信小程序遇上TensorFlow:Server端实现>中,我们探讨了微信小程序server端的实现.今天在调试微信小程序时才发现一个问题,那就是:微信小程序要求HTTP ...
Tensorflow：在Tensorflow的不同版本中如何实现Xavier参数权重初始化
Tensorflow:在Tensorflow的不同版本中如何实现Xavier参数权重初始化目录在Tensorflow的不同版本中实现Xavier参数权重初始化 Tensorflow1版本中实现
Tensorflow：tf.contrib.rnn.DropoutWrapper函数(谷歌已经为Dropout申请了专利！)、MultiRNNCell函数的解读与理解
Tensorflow:tf.contrib.rnn.DropoutWrapper函数(谷歌已经为Dropout申请了专利!).MultiRNNCell函数的解读与理解目录 1.tf.contrib. ...
DL框架之TensorFlow：深度学习框架TensorFlow Core(低级别TensorFlow API)的简介、安装、使用方法之详细攻略
DL框架之TensorFlow:TensorFlow Core(低级别TensorFlow API)的简介.安装.使用方法之详细DL框架之TensorFlow:深度学习框架TensorFlow Cor ...

Debug Tensorflow ：Two checkpoint references resolved to different objects

环境

报错

解决方案

参考

Debug Tensorflow ：Two checkpoint references resolved to different objects相关推荐

最新文章

热门文章