一、Base
- 使用这个分类网络,学习率固定为1e-4,优化方法是SGD,损失函数是CrossEntropyLoss
- pytorch1.6
- 最高准确率为
0.794
,会出现过拟合
二、修改网络结构
- 1.每个卷积层之后加relu。结果是最高准确率为
0.819
,会出现过拟合 - 2.将relu全改成leakyrelu,添加权重初始化。结果和1差不多,没多久我打断了
- 3.在2的基础上,每个卷积层之后加BN,卷积层不使用bias。结果是最高准确率为
0.867
,不会出现过拟合,测试loss收敛于0.67~0.7