日韩性视频-久久久蜜桃-www中文字幕-在线中文字幕av-亚洲欧美一区二区三区四区-撸久久-香蕉视频一区-久久无码精品丰满人妻-国产高潮av-激情福利社-日韩av网址大全-国产精品久久999-日本五十路在线-性欧美在线-久久99精品波多结衣一区-男女午夜免费视频-黑人极品ⅴideos精品欧美棵-人人妻人人澡人人爽精品欧美一区-日韩一区在线看-欧美a级在线免费观看

歡迎訪問 生活随笔!

生活随笔

當前位置: 首頁 > 编程资源 > 编程问答 >内容正文

编程问答

Error(s) in loading state_dict for ResNet 问题解决

發布時間:2023/12/15 编程问答 33 豆豆
生活随笔 收集整理的這篇文章主要介紹了 Error(s) in loading state_dict for ResNet 问题解决 小編覺得挺不錯的,現在分享給大家,幫大家做個參考.

當使用如下命令加載模型的時候,會報錯,沒有一些 keys:

model.load_state_dict(model_zoo.load_url(model_urls['resnet101'])) RuntimeError: Error(s) in loading state_dict for ResNet:Missing key(s) in state_dict: "C1_down_channel.weight", "C1_down_channel.bias", "C2_down_channel.weight", "C2_down_channel.bias", "C3_down_channel.weight", "C3_down_channel.bias", "C4_down_channel.weight", "C4_down_channel.bias", "C5_down_channel.weight", "C5_down_channel.bias", "score_dsn1.weight", "score_dsn1.bias", "score_dsn2.weight", "score_dsn2.bias", "score_dsn3.weight", "score_dsn3.bias", "score_dsn4.weight", "score_dsn4.bias", "score_dsn5.weight", "score_dsn5.bias", "score_final.weight", "score_final.bias".

則可以使用如下命令來解決,因為 load 的模型可能會缺少一些現有模型需要的東西,不使用嚴格的方式來 load 即可:

model.load_state_dict(model_zoo.load_url(model_urls['resnet101']), strict=False)

總結

以上是生活随笔為你收集整理的Error(s) in loading state_dict for ResNet 问题解决的全部內容,希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯,歡迎將生活随笔推薦給好友。