resnet34 pytorch_pytorch参数初始化

pytorch是包含一些常见的神经网络模型的,ResNet34、ResNet18、VGG等等,都在models模块中,调用接口如下:

model = models.resnet34pretrained=True, progress=True)

预训练pretrained和进度条progress设为True或False就看你自己需求咯~

⚠️与此同时,你可能还会遇到pytorch官网的输出类别数与你所需不同:你想分七类,可resnet34官网是1000类,那就调整一下呗:

# 修改最后线性层的输出通道数
model.fc = nn.Linearmodel.fc.in_features, 9) # 想输出为9个类别时

Published by

风君子

独自遨游何稽首 揭天掀地慰生平

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注