floodsung / wechat_jump_end_to_end_train

Playing Wechat Jump Game with End-to-End Convolutional Neural Networks (Train code)

MIT License

45 stars 14 forks source link

readme

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