fix(classify-mnist): update torch packages (#14772)

Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
hongbo-miao · Feb 23, 2024 · 71ed9fc · 71ed9fc
1 parent 8ec602c
commit 71ed9fc
Show file tree

Hide file tree

Showing 4 changed files with 114 additions and 116 deletions.
diff --git a/machine-learning/hm-kubeflow/pipelines/classify-mnist/poetry.lock b/machine-learning/hm-kubeflow/pipelines/classify-mnist/poetry.lock
diff --git a/machine-learning/hm-kubeflow/pipelines/classify-mnist/pyproject.toml b/machine-learning/hm-kubeflow/pipelines/classify-mnist/pyproject.toml
@@ -7,8 +7,8 @@ authors = ["Hongbo Miao"]
 [tool.poetry.dependencies]
 python = "3.11.x"
 kfp = "2.7.0"
-torch = "2.2.0"
-torchvision = "0.17.0"
+torch = "2.2.1"
+torchvision = "0.17.1"
 lightning = "2.2.0.post0"
 
 [tool.poetry.group.dev.dependencies]