Add torch > req.txt

chizuchizu · Mar 25, 2021 · 5756ee2 · 5756ee2
1 parent 8996e85
commit 5756ee2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,7 @@ pandas==1.2.2
 matplotlib==3.3.4
 seaborn==0.11.1
 scikit-learn==0.24.1
-# torch==1.8.0
+torch==1.8.0
 opencv-python==4.5.1.48
 tqdm==4.58.0
 amplify[extra]