Skip to content

Commit

Permalink
replace the new step_delay function with delay(x,1), which worked pre…
Browse files Browse the repository at this point in the history
…viously for recurrent nets

Signed-off-by: Jonathan Timcheck <jonathan.timcheck@intel.com>
  • Loading branch information
timcheck committed Jun 12, 2024
1 parent 0b71df6 commit 543a95c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/lava/lib/dl/slayer/block/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -1478,7 +1478,7 @@ def forward(self, x):
self.spike_state = spike.clone().detach().reshape(z.shape[:-1])

if self.delay_shift is True:
x = step_delay(self, x)
x = delay(x, 1)
if self.delay is not None:
x = self.delay(x)

Expand Down

0 comments on commit 543a95c

Please sign in to comment.