rscup2019,分类赛道 baseline
0.893
pytorch==1.1 (1.0应该也可以)
tensorboard==1.8
tensorboardX
pillow
8G内存,1080Ti,更弱一点的显卡也可以,注意调低batch_size参数
最简单的使用方法,只需要指明data_dir参数即可,90分钟之后,就可以得到最终模型以及log、tensorboard_log了
python train_resnet.py --data_dir your_data_dir
data_dir 之下的目录结构应如下:
data_dir/
|->train
|->val
|->test
|->ClsName2id.txt