python  train.py 50000 300 256 0.3 0.3 400 checkpoints/
python  predict.py 50000 300 256 0.3 0.3 400 checkpoints/