diff --git a/Dockerfile b/Dockerfile index e70e496..6577653 100644 --- a/Dockerfile +++ b/Dockerfile @@ -19,6 +19,7 @@ RUN pip3 install pymongo RUN pip3 install sacred RUN pip3 install mlflow RUN pip3 install torch +RUN pip3 install dvc CMD python3 data_expl.py CMD python3 nn_train.py \ No newline at end of file