在浏览器中完全运行经过训练的PyTorch LSTM/GRU模型的最佳方式

Question

作为交互式博客文章的一部分，我正在考虑在浏览器中完全运行经过训练的PyTorch模型(包含LSTM/GRU层(。我看过ONNX.js，它非常有效，但不适用于包含GRU层的模型。我在ONNX.js github上看到有人评论说Gated RNN还不受支持，但那是半年多前的事了，我找不到任何其他信息。

除此之外，最好的选择似乎是在Tensorflow中重写模型并导出到Tensorflow.js。

有没有更简单、更直接的解决方案？

Answer 1

有一个线程，它描述了选项，但没有受到太多关注。

总之，截至2020年5月，只有两种选择：

1( ONNX.js，但其开发目前已经过时。

2( 将模型转换为Tensorflow。

从技术上讲，还有第三种，不包括服务器。这就是在移动应用程序中运行该模型。