pytorch是包含一些常見的神經網絡模型的，ResNet34、ResNet18、VGG等等，都在models模塊中，調用接口如下：

model = models.resnet34(pretrained=True, progress=True)

預訓練pretrained和進度條progress設為True或False就看你自己需求咯～

⚠️與此同時，你可能還會遇到pytorch官網的輸出類別數與你所需不同：你想分七類，可resnet34官網是1000類，那就調整一下唄：

# 修改最后線性層的輸出通道數
model.fc = nn.Linear(model.fc.in_features, 9) # 想輸出為9個類別時

總結

以上是生活随笔為你收集整理的resnet34 pytorch_pytorch参数初始化的全部內容，希望文章能夠幫你解決所遇到的問題。

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