Specify torch version, which fixed #199

Should fix an error which occurs sporadically when training.
graphnet-team · May 17, 2022 · 764dc87 · 764dc87
1 parent 03f48cb
commit 764dc87
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -22,6 +22,7 @@ def install(package):
     "numpy",
     "timer",
     "tqdm",
+    "torch==1.10.1",
     "torch-cluster==1.5.9",
     "torch-scatter==2.0.9",
     "torch-sparse==0.6.12",