问题描述
有没有办法处理包含循环的注意力解码器块?实现在此 Colab Link
问题
我希望导出一个包含解码循环的注意块。 由于此 issue,LSTMCell 似乎无法导出到 ONNX。因此,我通过采用单个时间步进行推理来使用 LSTM 工作,例如 LSTMCell。我该如何处理这种模型?这是pytorch中onnx导出功能的问题吗?
如果我使用 use_external_data_format=True 选项导出,它不会保存到单个文件,而是保存到许多分离的文件中。
我在 ONNX github issue 中打开了一个问题并在 stackoverflow 中询问,因为我不确定这是错误/功能问题还是问题。
备注
似乎以下导出非常大图的实现无法导出到 ONNX 并显示以下错误。
RuntimeError: Exporting model exceed maximum protobuf size of 2GB. Please call torch.onnx.export with use_external_data_format=True.
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)