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()

 

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