rgrainger commited on
Commit
3d67e33
1 Parent(s): 71f183c

adding requirements

Browse files
Files changed (2) hide show
  1. .gitignore +1 -2
  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