2018广东工业智造大数据创新大赛——智能算法赛,未调参情况下线上0.921
|--data
|--guangdong_round1_train1_20180903
|--guangdong_round1_train2_20180916
|--guangdong_round1_test_a_20180916
|--gen_label_csv.py
|--model_v4.py
|--main_inception_v4.py
python gen_label_csv.py
python main_inception_v4.py
框架:Pytorch 0.4
代码经测试,线上分数为0.921
,线上19名(截至2018.9.20)。
如果有版本迭代,只需修改主程序名,则代码会自动生成新目录用于保存模型与结果文件。
使用的模型为ImageNet预训练的inception v4,整体思路:数据增强之后直接将网络全部层一起训练,做简单分类。
本代码为baseline,未经任何调参,训练参数和数据增强方式等可以自行随意修改。
输入尺寸为384×384,如果出现out of memory(显存不足),可减小batch size再重新跑。
注:只用到了guangdong_round1_train2_20180916数据,没有使用guangdong_round1_train1_20180903,可自行添加。
许多同学反应Windows下训练时train_loader报错的问题,将workers设为0即可解决。
https://tianchi.aliyun.com/forum/postDetail?spm=5176.12586969.1002.9.743c2881MOwYf2&postId=38469