OpenGVLab/VideoMamba

关于记录梯度值

776991400 opened this issue · 0 comments

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