diff --git a/dockerfiles/cuda120/Dockerfile b/dockerfiles/cuda120/Dockerfile
index d008540..2b112e1 100644
--- a/dockerfiles/cuda120/Dockerfile
+++ b/dockerfiles/cuda120/Dockerfile
@@ -14,4 +14,3 @@ COPY ./src /workdir/src
 WORKDIR /workdir
 
 ENTRYPOINT [ "python3", "-m" ]
-CMD [ "src.train" ]