Add requirements.txt

3 years ago · 536a65239d
parent a5999d406c
commit 536a65239d
1 changed files with 13 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1,13 @@
+torch==1.6.0
+torchvision==0.7.0
+tqdm==4.61.1
+termcolor==1.1.0
+addict==2.4.0
+opencv-python==4.5.2.54
+scikit-learn==0.24.2
+pytorch-warmup==0.0.4
+tensorboard==2.5.0
+pandas==1.1.5
+scipy==1.5.4
+pathspec==0.8.1
+yapf==0.31.0