Train on full set of images

cc: @subutai