Spaces:
Sleeping
Sleeping
adding requirements
Browse files- .gitignore +1 -2
- requirements.txt +16 -0
.gitignore
CHANGED
@@ -139,5 +139,4 @@ config-lock.yaml
|
|
139 |
*.p
|
140 |
datasets/*
|
141 |
data/*
|
142 |
-
*.save
|
143 |
-
*.txt
|
|
|
139 |
*.p
|
140 |
datasets/*
|
141 |
data/*
|
142 |
+
*.save
|
|
requirements.txt
ADDED
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
torch>=1.12.1
|
2 |
+
torchvision==0.13.1
|
3 |
+
pytorch-ignite==0.4.11
|
4 |
+
pyyaml
|
5 |
+
pytest
|
6 |
+
openmim==0.3.7
|
7 |
+
mmcv==2.0.0
|
8 |
+
mmpretrain==1.0.0rc7
|
9 |
+
mmsegmentation==1.0.0
|
10 |
+
timm==0.6.12
|
11 |
+
gradio==3.50.2
|
12 |
+
tqdm
|
13 |
+
cvxpylayers @ git+https://github.com/cvxgrp/cvxpylayers.git
|
14 |
+
qpth @ git+https://github.com/locuslab/qpth.git
|
15 |
+
torch_optimizer
|
16 |
+
scikit-learn
|