diff --git a/modneflib/modnef/modnef_torch/model.py b/modneflib/modnef/modnef_torch/model.py index 3b86faafa5ce0886c01220282104ac8b17ebf53a..bf63ae0a948894ad047788c1173bf4de8d5d171e 100644 --- a/modneflib/modnef/modnef_torch/model.py +++ b/modneflib/modnef/modnef_torch/model.py @@ -162,11 +162,11 @@ class ModNEFModel(nn.Module): """ if self.fpga_eval_flag: - return self._fpga_forward(input_spikes) + return self.fpga_forward(input_spikes) else: return self.software_forward(input_spikes) - def _fpga_forward(self, input_spikes): + def fpga_forward(self, input_spikes): """ Transmit input spike to FPGA @@ -181,8 +181,6 @@ class ModNEFModel(nn.Module): output_spike, None """ - print("hi") - def to_aer(input): input = input.reshape(-1).to(torch.int32)