解決用gpu訓練好的神經網絡,用tensorflow-cpu跑報錯問題

訓練的時候當然用gpu,速度快呀。 我想用cpu版的tensorflow跑一下,結果報錯,這個錯誤不太容易看懂。 大概意思是沒找到一些節點。 後來發現原因,用gpu和cpu保存的pb模型不太一樣,但是checkpoints文件是通用的。 使用tensorflow-cpu再把checkpoints文件重新轉換一下pb文件就可以了。 完美解決!
相關文章
相關標籤/搜索