Browse Source

fix bug

master
fangwei123456 1 week ago
parent
commit
bfdedfee61
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      spikingjelly/clock_driven/layer.py

+ 2
- 2
spikingjelly/clock_driven/layer.py View File

@@ -865,9 +865,9 @@ class SeqToANNContainer(nn.Module):
:rtype: torch.Tensor
'''
y_shape = [x_seq.shape[0], x_seq.shape[1]]
y_seq = self.module(x_seq.flatten(0, 1))
y_seq = self.module(x_seq.flatten(0, 1).contiguous())
y_shape.extend(y_seq.shape[1:])
return y_seq.reshape(y_shape)
return y_seq.view(y_shape)

class STDPLearner(nn.Module):
def __init__(self,


Loading…
Cancel
Save