请教GPU方面的问题

fangwei123456 / spikingjelly

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

https://spikingjelly.readthedocs.io

Other

1.35k stars 239 forks source link

Closed Met4physics closed 1 year ago

Met4physics commented 1 year ago

Issue type

SpikingJelly version

latest

Description 请问你们显存占用也是非常多吗，我实现了一个unet，T=4时占9g，T=8直接把3090撑爆了 ...

fangwei123456 commented 1 year ago

BPTT的显存占用和T几乎是正比的，所以占显存大很正常

Met4physics commented 1 year ago

好的感谢