微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决无法让 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 举报,一经查实,本站将立刻删除。