Update requirements.txt for coremltools kmeans quantization

2020-08-22 20:50:53 -07:00 · 2020-08-22 20:50:53 -07:00 · 5f07782f56
parent 71209a6099
commit 5f07782f56
1 changed files with 2 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -17,9 +17,10 @@ tqdm>=4.41.0
 # pycocotools>=2.0

 # export --------------------------------------
+# packaging  # for coremltools
 # coremltools==4.0b3
 # onnx>=1.7.0
-# packaging  # for coremltools
+# scikit-learn==0.19.2  # for coreml quantization

 # extras --------------------------------------
 # thop