admin管理员组

文章数量:1636810

RuntimeError: You must implement the backward function for custom autograd.Function.

问题原因

问题解决

主要原因是自己写的loss梯度 回传函数有问题

参考一下这一篇blog学习了一下怎么写自己的loss和网络梯度回传

一开始的错误代码是这样的

没有加上@staticmethod

加上后就可以用了

本文标签: 代码RuntimeErrorDebugimplementfunction