問題タブ [tf.keras]
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.
1831 問題
0
投票する
5
に答える
6469
参照
python - テンソルフロー2.0でケラスを使用して大規模なモデルを保存/復元する方法は?
新しい tensorflow 2.0 で作成され、keras と tensorflow を混合した大規模なカスタム モデルがあります。保存したい (アーキテクチャとウェイト)。再現する正確なコマンド:
私はモデルを保存することに成功しました：
しかし、それをロードしようとすると:
終了することはありません (エラー メッセージは表示されません)。モデルが大きすぎるのではないでしょうか？model.to_json()完全な API と同様にJSON として保存しようとしましtf.keras.models.save_model()たが、同じ問題があり、ロードできません (または少なくとも長すぎます)。
Windows/Linux で、GPU の有無にかかわらず同じ問題が発生します。
保存と復元は、完全な Keras と単純なモデルでうまく機能します。
編集
- 重みを保存してからロードすることはうまく機能しますが、モデル構造をロードすることは不可能です。
- バグを再現するために使用するモデルを置きます。これは Pix2Pix の例から来ています ( https://www.tensorflow.org/alpha/tutorials/generative/pix2pix )
- tensorflow github に関する問題も書きました: https://github.com/tensorflow/tensorflow/issues/28281