diff --git a/Dockerfile b/Dockerfile
index 52c0706..248e76d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,3 +18,5 @@ RUN pip3 install matplotlib
 RUN pip3 install --no-cache-dir tensorflow
 
 CMD ./run.sh
+CMD ./run_training.sh
+