From 7226c5de12c77aeb0d934c9d5dff62ba5fceed67 Mon Sep 17 00:00:00 2001 From: Hananel Hazan Date: Tue, 28 Jul 2020 14:00:10 -0400 Subject: [PATCH] Update BindsNET requirements to the new PyTorch version --- bindsnet/network/topology.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bindsnet/network/topology.py b/bindsnet/network/topology.py index eb44627f..2432011f 100644 --- a/bindsnet/network/topology.py +++ b/bindsnet/network/topology.py @@ -408,7 +408,7 @@ def compute(self, s: torch.Tensor) -> torch.Tensor: decaying spike activation). """ self.firing_rates -= self.decay * self.firing_rates - self.firing_rates += s.float() + self.firing_rates += s.float().squeeze() _, indices = F.max_pool2d( self.firing_rates,