tensorflow 使用VGG16模型VGG_16.tfmodel 報錯 GraphDef cannot be larger than 2GB.

https://hehuihui1994.github.io/hehuihui1994.github.io/2017/07/19/TensorFlow常見問題/

爆掉應該是因爲所有的變量都加載到了默認圖裏。

可能因爲我跑完第一個model之後,再跑第二個model,我以爲會覆蓋的,但是第一個model中的變量還在圖中…

 

解決方法:

在跑之前清楚以下默認圖。

#清空默認圖
tf.reset_default_graph()

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章