Wechat_Jump_End_to_End_Train
Prerequisite
pytorch
How to use it
Download the dataset from Baidu Yun
Place the dataset in this folder, then
python train.py
I train this model on Ubuntu 16.04 with GTX 970
Train vgg version and attention version
Almost the same with the basic version. Just try it.