记录jit编译的PyTorch类方法（Sphinx）

我在尝试用Sphinx记录自定义PyTorch模型时遇到了一个问题：jit编译的方法在文档中显示时没有docstring。我该如何解决这个问题？我查过了，但提议的解决方案似乎不起作用。当我尝试使用

.automethod

时，我得到

AttributeError: '_CachedForward' object has no attribute '__getattr__'

这里有一个MWE；目前，我通过编写一个

my_forward

并在

forward

中调用它来规避这个问题

来自火炬导入jit、张量的

类DummyModel（jit.ScriptModule）：
“我的虚拟模型”
定义初始化（自身，常数：浮点）：
超级（DummyModel，self）。\uuuu init\uuuu（）
self.const=张量（const）
@jit.script\u方法
def forward（自我，x:张量）->张量：
“”“此方法在文档中显示为：undoc成员”“”
返回自我。我的_向前（x）
定义my_forward（自我，x:张量）->张量：
“”“此方法在文档中显示为：成员：”“”
返回x+self.const
运行Sphinx时，将环境变量设置为0。这将禁用脚本和跟踪注释（装饰器）

看。
我对派托克知之甚少。运行Sphinx时设置PYTORCH_JIT=0是否有帮助@mzjn确实做到了，非常感谢！我建议你把它写成真实的答案，这样我才能接受。