環境

• 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的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。