summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--requirements.txt3
1 files changed, 3 insertions, 0 deletions
diff --git a/requirements.txt b/requirements.txt
index e69de29..46fb677 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -0,0 +1,3 @@
+torch~=1.7.1
+torchvision~=0.8.0a0+ecf4e9c
+numpy~=1.19.4 \ No newline at end of file