Я пытаюсь запустить обученную модель PyTorch (содержащую слои LSTM / GRU) полностью в браузере (без серверной части) как часть интерактивного сообщения в блоге. Я посмотрел на ONNX. js, и это прекрасно работает, но не для модели, содержащей слой GRU. Я видел, как кто-то комментирует на ONNX. js github, что Gated RNN еще не поддерживаются, но это более полугода go, и я не могу найти никакой другой информации об этом.
Кроме этого, похоже, что лучшим вариантом будет просто переписать модель в Tensorflow и экспортировать в Tensorflow. js.
Есть ли более простое и прямое решение?