为了修改网络,我们重新初始化Conv2d层,使输出特征图深度为2 model.classifier[1] = nn.Conv2d(512, num_classes, kernel_size=(1,1), stride=(1,1)) ...
確定! 回上一頁