已开启
[fully_shard]注册PostBackwardHook时,输入requires_grad属性跳变 #66
MengXY107创建于  3月31日
MengXY107
MengXY107成员
3月31日 创建

该问题是怎么引起的?

在注册PostBackwardHook时,调用Autograd.Function输入中,既有需要求导的非叶子节点,也有不需要求导的非叶子节点,生成的输出中所有非叶子节点的requires_grad全变成True,不符合预期

重现步骤

报错信息

likedislike
MengXY107MengXY107成员
3月31日 将 Meng107 设为负责人