Init

adap · Jan 7, 2025 · b497a15 · b497a15
1 parent 53bd515
commit b497a15
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/baselines/flwr_baselines/pyproject.toml b/baselines/flwr_baselines/pyproject.toml
@@ -39,8 +39,8 @@ classifiers = [
 python = ">=3.8.15, <=3.11.0"
 # Mandatory dependencies
 flwr = { extras = ["simulation"], version = "^1.3.0" }
-torch = "^1.10.1"
-torchvision = "^0.11.2"
+torch = "2.5.1"
+torchvision = "0.20.1"
 hydra-core = "^1.2.0"
 numpy = "^1.20.0"
 tqdm = "4.66.3"