問題タブ [state-dict]
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.
9 問題
0
投票する
1
に答える
3851
参照
python - RuntimeError: Generator の state_dict の読み込み中にエラーが発生しました: state_dict にキーがありません
MNIST データセットを使用して DCGAN モデルをトレーニングしようとしていましたが、gen.state_dict()トレーニングが終了した後にロードできません。
このステップで gen state_dict を読み込めません:
エラーは次のとおりです。
0
投票する
1
に答える
32
参照
pytorch - 関数内の state_dict/checkpoint の保存について (PyTorch)
model_state チェックポイントを保存するために、次の関数を実装しようとしています。
以前は、for ループを実行するだけで同じことを行いました。
checkpoint={}ループごとに開始して更新する方法はありますか? またはcheckpoint={}、モデル自体が を保持しているため、すべてのエポックで問題ありませんstate_dict()。毎回チェックポイントを上書きしているだけです。