Skip to content
Snippets Groups Projects
Commit 7c7480bf authored by ahoni's avatar ahoni
Browse files

remove post update quantization

parent 33502b1d
Branches
No related tags found
1 merge request!3Dev
...@@ -7,7 +7,6 @@ Dependencies: uart_1step, uart_classifier, uart_classifier_timer, uart_xstep, ua ...@@ -7,7 +7,6 @@ Dependencies: uart_1step, uart_classifier, uart_classifier_timer, uart_xstep, ua
Descriptions: UART module builder init Descriptions: UART module builder init
""" """
from .uart_1step import Uart_1Step
from .uart_classifier import Uart_Classifier from .uart_classifier import Uart_Classifier
from .uart_classifier_timer import Uart_Classifier_Timer from .uart_classifier_timer import Uart_Classifier_Timer
from .uart_xstep import Uart_XStep from .uart_xstep import Uart_XStep
......
...@@ -11,6 +11,5 @@ from .classifier_driver import Classifier_Driver ...@@ -11,6 +11,5 @@ from .classifier_driver import Classifier_Driver
from .classifier_timer_driver import Classifier_Timer_Driver from .classifier_timer_driver import Classifier_Timer_Driver
from .debugger_driver import Debugger_Driver from .debugger_driver import Debugger_Driver
from .default_driver import ModNEF_Driver from .default_driver import ModNEF_Driver
from .single_step_driver import SingleStep_Driver
from .xstep_driver import XStep_Driver from .xstep_driver import XStep_Driver
from .xstep_timer_driver import XStep_Timer_Driver from .xstep_timer_driver import XStep_Timer_Driver
\ No newline at end of file
...@@ -13,7 +13,6 @@ from .drivers import * ...@@ -13,7 +13,6 @@ from .drivers import *
import yaml import yaml
drivers_dict = { drivers_dict = {
"1Step" : SingleStep_Driver,
"XStep" : XStep_Driver, "XStep" : XStep_Driver,
"Classifier" : Classifier_Driver, "Classifier" : Classifier_Driver,
"Debugger" : Debugger_Driver, "Debugger" : Debugger_Driver,
......
...@@ -269,9 +269,6 @@ class ShiftLIF(LIF, ModNEFNeuron): ...@@ -269,9 +269,6 @@ class ShiftLIF(LIF, ModNEFNeuron):
else: else:
self.mem = self.mem-self.__shift(self.mem) 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: if self.hardware_estimation_flag:
self.val_min = torch.min(torch.min(input_.min(), self.mem.min()), self.val_min) 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) self.val_max = torch.max(torch.max(input_.max(), self.mem.max()), self.val_max)
......
...@@ -69,7 +69,7 @@ def train(model, trainLoader, testLoader, optimizer, loss, device=torch.device(" ...@@ -69,7 +69,7 @@ def train(model, trainLoader, testLoader, optimizer, loss, device=torch.device("
acc_test_history.append(acc_test) acc_test_history.append(acc_test)
if best_model_name!="" and acc_test>best_acc: 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 best_acc = acc_test
if save_history: if save_history:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment