Dockerfile to run training