您好,登錄后才能下訂單哦!
這篇文章主要為大家展示了“PyTorch如何使用預(yù)訓(xùn)練模型”,內(nèi)容簡而易懂,條理清晰,希望能夠幫助大家解決疑惑,下面讓小編帶領(lǐng)大家一起研究并學(xué)習(xí)一下“PyTorch如何使用預(yù)訓(xùn)練模型”這篇文章吧。
使用預(yù)訓(xùn)練模型的代碼如下:
# 加載預(yù)訓(xùn)練模型 resNet50 = models.resnet50(pretrained=True) ResNet50 = ResNet(Bottleneck, [3, 4, 6, 3], num_classes=2) # 讀取參數(shù) pretrained_dict = resNet50.state_dict() model_dict = ResNet50.state_dict() # 將pretained_dict里不屬于model_dict的鍵剔除掉 pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict} # 更新現(xiàn)有的model_dict model_dict.update(pretrained_dict) # 加載真正需要的state_dict ResNet50.load_state_dict(model_dict)
以上是“PyTorch如何使用預(yù)訓(xùn)練模型”這篇文章的所有內(nèi)容,感謝各位的閱讀!相信大家都有了一定的了解,希望分享的內(nèi)容對(duì)大家有所幫助,如果還想學(xué)習(xí)更多知識(shí),歡迎關(guān)注億速云行業(yè)資訊頻道!
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場,如果涉及侵權(quán)請(qǐng)聯(lián)系站長郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。