問題タブ [tensorflowjs-converter]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
javascript - tensorflowJS のステートフル LSTM: エラー: Tensor が破棄されました
TensorflowJS を使用してステートフル LSTM セルを含むモデルをデプロイしようとしています。モデルは keras で事前トレーニングされており、次のように構築されています。
モデルをトレーニングした後、tensorflowjs_converterで変換し、 tf.loadLayersModel("/fullmodelTF/model.json").
予測をしようとすると、 の最初の呼び出しで機能しmodel.predict()ます。もう一度予測しようとすると、次のエラーが発生します。
開発者コンソールでモデル オブジェクトを調べたところ、ステートフル LSTM レイヤーでプロパティstates_が次のようになっていることに気付きました。
ご覧のとおり、内部状態の 1 つが破棄されました。この破棄と結果として生じるエラーを回避するにはどうすればよいですか?
同じモデルをステートレスとしてデプロイしようとすると、問題なく動作します。
JavaScript での予測のループは次のようになります。