From 56467ab8fcb9d757a4b3fdc0dfb8da1d1b9f43e7 Mon Sep 17 00:00:00 2001
From: ptresson <paul.tresson@ird.fr>
Date: Thu, 17 Oct 2024 09:47:50 +0200
Subject: [PATCH] fix quantization
---
encoder.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/encoder.py b/encoder.py
index c888b22..17ec23c 100644
--- 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 :
--
GitLab