fix quantization

56467ab8 · paul.tresson_ird.fr · 868cb720 · 56467ab8
Commit 56467ab8 authored 6 months ago by paul.tresson_ird.fr
--- a/encoder.py
+++ b/encoder.py
@@ -440,7 +440,7 @@ class EncoderAlgorithm(QgsProcessingAlgorithm):
            try :
                feedback.pushInfo(f'before quantization : {get_model_size(model)}')

-                quantize_model(model, device)
+                model = quantize_model(model, device)
                feedback.pushInfo(f'after quantization : {get_model_size(model)}')

            except :