Update readme & refactor train.sh

3 jobs for master in 13 minutes and 7 seconds (queued for 3 seconds)
Status Job ID Name Coverage
  Check Style
passed #4927
pep8

00:00:28

 
  Build
passed #4928
build_cpu_image

00:07:50

passed #4929
build_gpu_image

00:12:38