commit

LaurinmyReha · Nov 11, 2021 · 6b36bc2 · 6b36bc2
1 parent dd9632f
commit 6b36bc2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/experimental/torchprep/README.md b/experimental/torchprep/README.md
@@ -39,8 +39,12 @@ torchprep profile models/resnet152.pt --iterations 100 --device cpu --input-shap
 
 # set omp threads to 1 to optimize cpu inference
 torchprep env --device cpu
+
+# Prune 30% of model weights
+torchprep prune models/resnet152.pt --prune_amount 0.3
 ```
 
+
 ### Available commands
 
 
@@ -64,6 +68,8 @@ Commands:
 
 ### Usage instructions for a command
 
+`torchprep <command> --help`
+
 ```
 Usage: torchprep quantize [OPTIONS] MODEL_PATH PRECISION:{int8|float16}