jeshraghian / snntorch

Deep and online learning with spiking neural networks in Python
https://snntorch.readthedocs.io/en/latest/
MIT License
1.17k stars 202 forks source link

Small error in tutorial 7 #324

Closed bjourne closed 1 month ago

bjourne commented 1 month ago

The event tensors' dimension changed to torch.Size([128, 309, 2, 34, 34]) in the latest version. Easy to fix by rearranging the dimensions like this: targets = torch.movedim(targets, 0, 1)

bjourne commented 1 month ago

I misread the text. Mea culpa