作为交互式博客文章的一部分，我正在考虑在浏览器(无后端)中完全运行经过训练的 PyTorch 模型(包含 LSTM/GRU 层)。我看过 ONNX.js，这很好用，但不适用于包含 GRU 层的模型。我在 ONNX.js github 上看到有人评论说 Gated RNN 尚不受支持，但那是半年前的事了，我找不到关于此的任何其他信息。
除此之外，似乎最好的选择是在 Tensorflow 中重写模型并导出到 Tensorflow.js。
有没有更简单、更直接的解决方案？
最佳答案
有这个thread ，它描述了选项，但没有受到很多关注。
综上所述，截至2020年5月，只有两种选择:
1) ONNX.js 但它的开发目前是 stale .
2) 将模型转换为 Tensorflow。
从技术上讲，还有第三个，不包括服务器。那就是在 Mobile Application 中运行模型.
关于javascript - 在浏览器中完全运行经过训练的 PyTorch LSTM/GRU 模型的最佳方式，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/60340552/