Skip to content

Commit 7226c5d

Browse files
committed
Update BindsNET requirements to the new PyTorch version
1 parent 2cfaf8a commit 7226c5d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bindsnet/network/topology.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,7 @@ def compute(self, s: torch.Tensor) -> torch.Tensor:
408408
decaying spike activation).
409409
"""
410410
self.firing_rates -= self.decay * self.firing_rates
411-
self.firing_rates += s.float()
411+
self.firing_rates += s.float().squeeze()
412412

413413
_, indices = F.max_pool2d(
414414
self.firing_rates,

0 commit comments

Comments
 (0)