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.

Reference

https://github.com/wangshub/wechat_jump_game