cannot register a hook on a tensor that doesn‘t require gradient的解决

仅作为记录，大佬请跳过。

博主在explan-tran时，新建该模型（用训练集训练和测试集测试），在用测试集测试时，出现cannot register a hook on a tensor that doesn't require gradient的问题。

经pychram调试看参数，发现是在原模型设计时，有保存梯度的步骤；而博主的程序在测试是，不用梯度，因此出现模型的保存错误

做法

将原模型中保存梯度的代码注释掉，即可。

展示

就可以连续运行了：