OpenGVLab / VideoMamba

VideoMamba: State Space Model for Efficient Video Understanding
https://arxiv.org/abs/2403.06977
Apache License 2.0
660 stars 47 forks source link

关于记录梯度值 #58

Open 776991400 opened 4 weeks ago

776991400 commented 4 weeks ago

你好,我想请问一下,我注册钩子来记录梯度值,当把钩子注册在mixer层时有正常的梯度值,但是进一步注册在mixer.conv1d就是None了。这是什么原因造成的?