diff --git a/containers/Dockerfile b/containers/Dockerfile
index 906e2659..0d27cc8d 100644
--- a/containers/Dockerfile
+++ b/containers/Dockerfile
@@ -27,12 +27,13 @@ ENV LC_ALL=C
 RUN pip install --upgrade pip
 #RUN pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
 RUN ls 
-RUN pip install -r env-files/torch/pytorch-env-gpu-container.txt
+#RUN pip install -r env-files/torch/pytorch-env-gpu-container.txt
 RUN pip install -e .
 
 # Define the run command when the container starts
-CMD ["sleep", "300"]
-CMD ["python", "-V"]
+#CMD ["sleep", "300"]
+#CMD ["python", "-V"]
+CMD ["python", "train.py", "-p", "pipeline.yaml"]
 
 # Additional information labels
 LABEL Author="azoechba"