待解決的問題

通過 saver=tf.train.import_meta_graph('./my_ckpt_save_dir/wdy_model-15.meta') 是把以前的整個網絡加載到新圖裏面,如何只加載我需要的部分呢。 經過試驗發現,經過finetune之後的網絡,還是會保留以前的結構。   如果要繼續訓練之前的部分,需要注意優化器的名字別重名了,重新命名下。圖下面有各個節點的名字可供查看。
相關文章
相關標籤/搜索