Fix pytorch env path (#55)

* Fix path * Update --------- Co-authored-by: Matteo Bunino <[email protected]>
interTwin-eu · Aug 17, 2023 · ccb5cf8 · ccb5cf8
1 parent f0e407b
commit ccb5cf8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/use-cases/mnist/mnist-ai-train.yml b/use-cases/mnist/mnist-ai-train.yml
@@ -11,6 +11,7 @@ train:
 
     # Lightning Trainer configuration
     trainer:
+      # Set to "cpu" when using pytorch "cpuonly" package
       accelerator: auto
       strategy: auto
       devices: auto

diff --git a/use-cases/mnist/training-workflow.yml b/use-cases/mnist/training-workflow.yml
@@ -20,7 +20,7 @@ steps:
       doc: Run MLFlow server on localhost
       command: python ${root}/mlflow-server.py
       env: 
-        file: ${ai-root}/env-files/pytorch-lock.yml
+        file: ${ai-root}/env-files/pytorch-gpu-lock.yml
         prefix: ${ai-root}/.venv-pytorch
         source: ${ai-root}
       args:
@@ -30,7 +30,7 @@ steps:
       doc: Train a neural network to classify MNIST images
       command: itwinai train
       env: 
-        file: ${ai-root}/env-files/pytorch-env-gpu.yml
+        file: ${ai-root}/env-files/pytorch-gpu-lock.yml
         prefix: ${ai-root}/.venv-pytorch
         source: ${ai-root}
       args:
@@ -41,7 +41,7 @@ steps:
       doc: Stop MLFlow server on localhost, if running
       command: python ${root}/mlflow-server.py
       env: 
-        file: ${ai-root}/env-files/pytorch-lock.yml
+        file: ${ai-root}/env-files/pytorch-gpu-lock.yml
         prefix: ${ai-root}/.venv-pytorch
         source: ${ai-root}
       args: