From 991c34139638931e9f84586b1c0dde695cce9eee Mon Sep 17 00:00:00 2001
From: ahoni <aurelie.saulq@proton.me>
Date: Mon, 10 Mar 2025 09:54:08 +0100
Subject: [PATCH] run lib fix
---
modneflib/modnef/modnef_torch/model.py | 4 +++-
modneflib/modnef/templates/run_lib.py | 2 +-
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/modneflib/modnef/modnef_torch/model.py b/modneflib/modnef/modnef_torch/model.py
index 0451bfe..3b86faa 100644
--- a/modneflib/modnef/modnef_torch/model.py
+++ b/modneflib/modnef/modnef_torch/model.py
@@ -181,6 +181,8 @@ class ModNEFModel(nn.Module):
output_spike, None
"""
+ print("hi")
+
def to_aer(input):
input = input.reshape(-1).to(torch.int32)
@@ -197,7 +199,7 @@ class ModNEFModel(nn.Module):
batch_result = []
for sample in input_spikes:
- sample_res = self.driver.run_sample(sample, to_aer, True, len(self.layers))
+ sample_res = self.driver.run_sample(sample, to_aer, True, 2) #TODO
batch_result.append([sample_res])
return torch.tensor(batch_result).permute(1, 0, 2), None
diff --git a/modneflib/modnef/templates/run_lib.py b/modneflib/modnef/templates/run_lib.py
index 5a98742..07ccb6d 100644
--- a/modneflib/modnef/templates/run_lib.py
+++ b/modneflib/modnef/templates/run_lib.py
@@ -202,7 +202,7 @@ def fpga_evaluation(model, testLoader, board_path, driver_config, name="FPGA Eva
return accuracy, y_pred, y_true
-def confusion_matrix(y_true, y_pred, file_name, classes):
+def conf_matrix(y_true, y_pred, file_name, classes):
cm = confusion_matrix(y_true, y_pred)
df_cm = pd.DataFrame(cm / np.sum(cm, axis=1)[:, None], index = [i for i in classes], columns = [i for i in classes])
plt.figure()
--
GitLab