Remove version specification of torch families

optuna · Oct 19, 2023 · 4079a69 · 4079a69
1 parent c57c82a
commit 4079a69
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pytorch/requirements.txt b/pytorch/requirements.txt
@@ -3,7 +3,7 @@ plotly
 pytorch-ignite
 pytorch-lightning
 skorch
-torch==2.0.0
-torchvision==0.15.1
-torchaudio==2.0.1
+torch
+torchvision
+torchaudio
 optuna