adding requirements.txt

benjaminjellis · benjaminjellis · commit eb1c38749ecc · 2021-02-20T18:37:01.000Z
diff --git a/RepNet/RepNetModel.py b/RepNet/RepNetModel.py
@@ -8,9 +8,7 @@
     conv feature extractor is different from that used in original paper
     check dilation on temporal 3d conv
     pairwise_l2_distance transpose for b
-    does get sims actually work
     input projection kernel_regularizer
-    transformer_layers_config dff needed ???
 """
 
 import torch
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,5 @@
+numpy==1.19.5
+Pillow==8.1.0
+torch==1.7.1
+torchvision==0.8.2
+typing-extensions==3.7.4.3
