无法让 torch.jit.script 与 pytorch RNN 一起正常工作

问题描述

好的，我需要编写 pytorch LSTM 脚本，因为当我进行跟踪时，它会通过不同的输入得到不同的答案。但是当我这样做时：

script_module=torch.jit.script(torch.nn.modules.rnn.LSTM(768,256,num_layers=2,batch_first=True,bidirectional=True))

生成的 script_module 打印为：RecursiveScriptModule(original_name=LSTM)，但与其他 recursivescriptmodules（例如，在此处生成：https://pytorch.org/tutorials/beginner/Intro_to_TorchScript_tutorial.html，并由我成功复制）不同，它没有 .code 或 .graph 属性，它不能被调用。

那我做错了什么？它显然“认为”它产生了一个正确的脚本。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）