Hello, I think there is a bug in:
_spikingjelly/activationbased/layer/BatchNorm1d
elif self.step_mode == 'm':
if x.dim() != 4 or x.dim() != 3:
raise ValueError(f'expected x with shape [T, N, C, L] or [T, N, C], but got x with shape {x.shape}!')
I think it should be an and instead of or. Because if x.dim is 4 then x.dim() != 3 is true and the error will be raised. Which is not wanted. Here is an example I got:
Issue type
SpikingJelly version
latest version
Description
Hello, I think there is a bug in: _spikingjelly/activationbased/layer/BatchNorm1d
I think it should be an
and
instead ofor
. Because if x.dim is 4 thenx.dim() != 3
is true and the error will be raised. Which is not wanted. Here is an example I got: