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