Maximef commited on
Commit
e056d91
·
1 Parent(s): cf2612a

add dependencies

Browse files
Files changed (1) hide show
  1. requirements.txt +23 -1
requirements.txt CHANGED
@@ -9,4 +9,26 @@ torch>=1.7.0,!=1.12.0
9
  torchvision>=0.8.1,!=0.13.0
10
  tqdm>=4.41.0
11
  protobuf<4.21.3
12
- pandas>=1.1.4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  torchvision>=0.8.1,!=0.13.0
10
  tqdm>=4.41.0
11
  protobuf<4.21.3
12
+ pandas>=1.1.4
13
+
14
+ # Logging -------------------------------------
15
+ tensorboard>=2.4.1
16
+ # wandb
17
+
18
+ # Plotting ------------------------------------
19
+ pandas>=1.1.4
20
+ seaborn>=0.11.0
21
+
22
+ # Export --------------------------------------
23
+ # coremltools>=4.1 # CoreML export
24
+ # onnx>=1.9.0 # ONNX export
25
+ # onnx-simplifier>=0.3.6 # ONNX simplifier
26
+ # scikit-learn==0.19.2 # CoreML quantization
27
+ # tensorflow>=2.4.1 # TFLite export
28
+ # tensorflowjs>=3.9.0 # TF.js export
29
+ # openvino-dev # OpenVINO export
30
+
31
+ # Extras --------------------------------------
32
+ ipython # interactive notebook
33
+ psutil # system utilization
34
+ thop # FLOPs computation