解決TensorFlow多次運行神經網絡衝突的問題

問題來源:爲了驗證神經網絡的穩定性,我把數據集隨機劃分了10次(每次都有訓練集和測試集),那怎麼讓神經網絡自動跑十次,得出平均準確率呢?即如何讓用tf搭建的神經網絡在多個數據集中依次訓練?

如果只是簡單的加for循環:

會導致如下錯誤:

ValueError: Variable InceptionV3/Conv2d_1a_3x3/weights already exists, disallowed. Did you mean to set reuse=True or reuse=tf.AUTO_REUSE in VarScope? Originally defined at:

因爲tf是有默認計算圖的,只能有一個,這樣等於啓動很多個計算圖,就會衝突報錯。

解決辦法:

tf.reset_default_graph()

加上這句語句就可以重新初始化,避免衝突,能成功多次運行神經網絡了。

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