Tensorflow: 如何加載神經網絡部分參數

轉載請標明出處:https://blog.csdn.net/SingingFisherhtml   需求以下:先用Resnet50在ImageNet上預訓練,最後一層輸出爲類別數量,設爲1000。而後將保存下來的參數遷移到PascalVoc上作調優訓練。函數 問題:因爲PascalVoc只有20類,因此Resnet50最後一層輸出要改成20。此時直接用tf.train.Saver()的restor
相關文章
相關標籤/搜索