Merge pull request #19 from kapilvgit/main

minist fixes
iSPIRT · Jun 12, 2024 · 0b88dff · 0b88dff
2 parents 314deb9 + 46a6b29
commit 0b88dff
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/scenarios/mnist/ci/Dockerfile.preprocess b/scenarios/mnist/ci/Dockerfile.preprocess
@@ -11,7 +11,7 @@ RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
 RUN python3.9 get-pip.py
 
 ## Install dependencies
-RUN pip3 install torch --index-url https://download.pytorch.org/whl/cpu
+RUN pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
 RUN pip3 --default-timeout=1000 install onnx onnx2pytorch
 
 RUN apt-get install -y jq

diff --git a/scenarios/mnist/ci/push-containers.sh b/scenarios/mnist/ci/push-containers.sh
@@ -0,0 +1,6 @@
+containers=("depa-mnist-save-model:latest" "depa-mnist-preprocess:latest")
+for container in "${containers[@]}"
+do
+  docker tag $container $CONTAINER_REGISTRY"/"$container
+  docker push $CONTAINER_REGISTRY"/"$container
+done