【第4次作业】猫狗大战挑战赛

猫狗大战挑战赛

关键代码

下载测试集,并把文件夹放到dogscats目录里一起格式化

我们这里使用已经训练好的模型。所以要冻结前面hidden layer的weight和bias，使用

model_vgg_new.classifier._modules['6'] = nn.Linear(4096, 2)

代码改变最后一层参数。这里的参数4096应该是最后一个hidden layer的neurons的数目，而这里的2就是最后的output layer，代表最后的分类结果只有两种。

在训练模型的过程中，我注意到这里没有使用softmax函数，如果在

        for inputs,classes in dataloader:

中的最后加入该函数，应该会使模型的loss更小。

预测结果