Dockerize the training so that it can easily be run on the newton cluster.
Dockerize the training so that it can easily be run on the newton cluster.