glenn-jocher commited on
Commit
5f07782
·
1 Parent(s): 71209a6

Update requirements.txt for coremltools kmeans quantization

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -17,9 +17,10 @@ tqdm>=4.41.0
17
  # pycocotools>=2.0
18
 
19
  # export --------------------------------------
 
20
  # coremltools==4.0b3
21
  # onnx>=1.7.0
22
- # packaging # for coremltools
23
 
24
  # extras --------------------------------------
25
  # thop
 
17
  # pycocotools>=2.0
18
 
19
  # export --------------------------------------
20
+ # packaging # for coremltools
21
  # coremltools==4.0b3
22
  # onnx>=1.7.0
23
+ # scikit-learn==0.19.2 # for coreml quantization
24
 
25
  # extras --------------------------------------
26
  # thop