出现在自定义求导函数时。有说是pytorch版本问题,需退到1.3之前。
或者按官方建议,在自定义的autorgrad.Function中的forward,backward前加上@staticmethod(似乎不加,按下面所说完成修改也可)
- 在调用时候,不再是实例化func()然后func(x),而是直接func.apply(x)
- 即使不加staticmethod,最后也是按静态方法调用的。staticmethod意味着Function不再能使用类内的方法和属性,去掉init()用别的方法代替
本文探讨了在使用PyTorch时遇到的自定义求导函数问题,提到可能需要降级到1.3版本,并强调了在forward和backward方法前添加@staticmethod的重要性。讲解了如何从实例化到apply调用方式的转变,以及静态方法对类内方法和属性的限制。
1万+

被折叠的 条评论
为什么被折叠?



