Fix bug in classification sample

l-bat · Aug 30, 2023 · b692d72 · b692d72
1 parent 9b10ef6
commit b692d72
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tools/pot/openvino/tools/pot/api/samples/classification/classification_sample.py b/tools/pot/openvino/tools/pot/api/samples/classification/classification_sample.py
@@ -210,10 +210,9 @@ def optimize_model(args):
     # Step 6: Execute the pipeline.
     compressed_model = pipeline.run(model)
 
-    # Step 7 (Optional): Compress model weights quantized precision
-    #                    in order to reduce the size of final .bin file.
-    if not args.keep_uncompressed_weights:
-        compress_model_weights(compressed_model)
+    # Step 7: Compress model weights quantized precision
+    #         in order to reduce the size of final .bin file.
+    compress_model_weights(compressed_model)
 
     return compressed_model, pipeline