Tensorflow2.0學習(17):利用resnet50進行遷移學習

利用預訓練好的resnet50模型進行遷移訓練 在下載resnet50模型時出現了域名解析錯誤,查了好多都不知道怎麼解決。 總的來看,網絡分爲了兩層:第一層是resnet50的模型,其中包含了不少卷積層與池化層等,而且已經預訓練好了權重偏置等。第二層是自定義的全鏈接層。在訓練模型時,設置的是僅訓練全鏈接層,及resnet50_fine_tune.layer[0].trainable = False
相關文章
相關標籤/搜索