Merge pull request #353 from masac11/master

fix bug of MultiDimensionalAttention function
fangwei123456 · Mar 23, 2023 · 3aee47e · 3aee47e
2 parents 9d94c32 + c65867d
commit 3aee47e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/spikingjelly/activation_based/layer.py b/spikingjelly/activation_based/layer.py
@@ -2131,6 +2131,7 @@ def forward(self, x):
         self.ca = ChannelAttention(C, reduction_c)
         self.sa = SpatialAttention(kernel_size)
         self.sigmoid = nn.Sigmoid()
+        self.relu = nn.ReLU()
 
     def forward(self, x: torch.Tensor):
         assert x.dim() == 5, ValueError(