惊蛰框架为什么在训练过程中会产生报错，是否可以提供调试的解决思路？

fangwei123456 / spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

https://spikingjelly.readthedocs.io

Other

1.4k stars 244 forks source link

Open zhangxiaodi opened 3 months ago

zhangxiaodi commented 3 months ago

具体说明：我的是pytorch 2.4.0 显卡 4090 问题是，在前面几个epoch都可以正常跑完，但是跑到十几个epoch的时候出现了这种问题。是否有调试和分析问题的思路呢？

Ym-Shan commented 3 months ago

最好提供下你的代码，是在某个epoch更改了输入的tensor尺寸吗？