pytorch-cifar10 对不同模型的测试

说明

代码链接：https://github.com/zhanghlgithub/pytorch-backbone-cifar10

为了方便快速学习和测试不同网络结构的性能表现，本模块封装好了接口。公开数据集基于cifar10数据测试。方便做实验和学习最新的方法。

使用方法

在使用时，只需把自己学到的最新网络结构添加到backbone模块即可，然后修改训练脚本train.sh中的backbone参数，使得模型训练时调用你写的backbone。简单好用！

训练

训练脚本

bash train.sh

关键参数说明

• lr：初试学习率
• backbones：选择不同的backbones
• st_epoch：开始epoch
• end_epoch：结束epoch
• optim：选择优化器
  注：具体参数含义，查看train.parse_args()参数说明

训练过程

adam算法：

lr	epoch
0.001 或 0.003