Update requirements.txt for coremltools kmeans quantization

burglarhobbit · Aug 23, 2020 · 9838ffa · 9838ffa
1 parent e10983c
commit 9838ffa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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