From 7c7480bf567989071aa5847af1b194c6fccc21d7 Mon Sep 17 00:00:00 2001
From: ahoni <aurelie.saulq@proton.me>
Date: Tue, 18 Mar 2025 05:50:14 +0100
Subject: [PATCH] remove post update quantization

---
 modneflib/modnef/arch_builder/modules/UART/__init__.py         | 1 -
 modneflib/modnef/modnef_driver/drivers/__init__.py             | 1 -
 modneflib/modnef/modnef_driver/modnef_drivers.py               | 1 -
 .../modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py | 3 ---
 modneflib/modnef/templates/run_lib.py                          | 2 +-
 5 files changed, 1 insertion(+), 7 deletions(-)

diff --git a/modneflib/modnef/arch_builder/modules/UART/__init__.py b/modneflib/modnef/arch_builder/modules/UART/__init__.py
index 1bf9604..3cbbe30 100644
--- a/modneflib/modnef/arch_builder/modules/UART/__init__.py
+++ b/modneflib/modnef/arch_builder/modules/UART/__init__.py
@@ -7,7 +7,6 @@ Dependencies: uart_1step, uart_classifier, uart_classifier_timer, uart_xstep, ua
 Descriptions: UART module builder init
 """
 
-from .uart_1step import Uart_1Step
 from .uart_classifier import Uart_Classifier
 from .uart_classifier_timer import Uart_Classifier_Timer
 from .uart_xstep import Uart_XStep
diff --git a/modneflib/modnef/modnef_driver/drivers/__init__.py b/modneflib/modnef/modnef_driver/drivers/__init__.py
index 0615fd9..1bd6c71 100644
--- a/modneflib/modnef/modnef_driver/drivers/__init__.py
+++ b/modneflib/modnef/modnef_driver/drivers/__init__.py
@@ -11,6 +11,5 @@ from .classifier_driver import Classifier_Driver
 from .classifier_timer_driver import Classifier_Timer_Driver
 from .debugger_driver import Debugger_Driver
 from .default_driver import ModNEF_Driver
-from .single_step_driver import SingleStep_Driver
 from .xstep_driver import XStep_Driver
 from .xstep_timer_driver import XStep_Timer_Driver
\ No newline at end of file
diff --git a/modneflib/modnef/modnef_driver/modnef_drivers.py b/modneflib/modnef/modnef_driver/modnef_drivers.py
index be6f4db..11e2cc3 100644
--- a/modneflib/modnef/modnef_driver/modnef_drivers.py
+++ b/modneflib/modnef/modnef_driver/modnef_drivers.py
@@ -13,7 +13,6 @@ from .drivers import *
 import yaml
 
 drivers_dict = {
-  "1Step" : SingleStep_Driver,
   "XStep" : XStep_Driver,
   "Classifier" : Classifier_Driver,
   "Debugger" : Debugger_Driver,
diff --git a/modneflib/modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py b/modneflib/modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py
index d4f58d3..9bb6d1d 100644
--- a/modneflib/modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py
+++ b/modneflib/modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py
@@ -269,9 +269,6 @@ class ShiftLIF(LIF, ModNEFNeuron):
     else:
       self.mem = self.mem-self.__shift(self.mem)
 
-    if self.quantization_flag:
-      self.mem.data = self.quantizer(self.mem.data, True)
-
     if self.hardware_estimation_flag:
       self.val_min = torch.min(torch.min(input_.min(), self.mem.min()), self.val_min)
       self.val_max = torch.max(torch.max(input_.max(), self.mem.max()), self.val_max)
diff --git a/modneflib/modnef/templates/run_lib.py b/modneflib/modnef/templates/run_lib.py
index 07ccb6d..21f536b 100644
--- a/modneflib/modnef/templates/run_lib.py
+++ b/modneflib/modnef/templates/run_lib.py
@@ -69,7 +69,7 @@ def train(model, trainLoader, testLoader, optimizer, loss, device=torch.device("
     acc_test_history.append(acc_test)
 
     if best_model_name!="" and acc_test>best_acc:
-      torch.save(model.state_dict(), best_model_name)
+      torch.save(model.state_dict(), f"{output_path}/{best_model_name}")
       best_acc = acc_test
 
   if save_history:
-- 
GitLab