From 5f07782f569324bf32dbc63441128d4935c9254a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 22 Aug 2020 20:50:53 -0700 Subject: [PATCH] Update requirements.txt for coremltools kmeans quantization --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index f59809f..7e62ccf 100755 --- 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