refactor(baselines:skip) Update torch version if FjORD baseline (#…

…4790)
adap · Jan 7, 2025 · e532e1a · e532e1a
1 parent 1595907
commit e532e1a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/baselines/fjord/requirements.txt b/baselines/fjord/requirements.txt
@@ -2,6 +2,6 @@ coloredlogs==15.0.1
 hydra-core==1.3.2
 flwr[simulation]==1.5.0
 omegaconf==2.3.0
-torch==2.0.1
-torchvision==0.15.2
+torch==2.2.0
+torchvision==0.17.0
 tqdm==4.65.0