使用pytorch進行遷移學習 修改預訓練權重模型的分類層/輸出類別(附mmaction相關實現代碼)

文章目錄 example step1 文件創建 step2 得到預訓練模型 step3 模型修改 參考文章 example 以mmaction中的tsn+resnet50+kinetics訓練的模型做爲預訓練模型,拿來訓練TSN+resnet+ucf101。python step1 文件創建 創建文件tools/transfer_model.py。git step2 得到預訓練模型 在mmacti
相關文章
相關標籤/搜索