温婉小海豚

文章
4
资源
1
加入时间
3年0月9天

pytorch 修改预训练模型(全连接层、单个卷积层、多个卷积层)1. 修改全连接层类别数2. 修改某一层卷积3. 修改某几层卷积

1. 修改全连接层类别数model = torchvision.models.resnet50(pretrained=True)# 重定义最后一层model.fc = nn.Linear(2048,10)print(model.fc)2. 修改某一层卷积model = torchvision.models.resnet50(pretrained=True)# 重定义第一层卷积的输入通道数model.conv1 = nn.Conv2d(4, 64, kernel_size=7, stride