問題タブ [trax]
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.
19 問題
0
投票する
0
に答える
13
参照
python - モデルのトレーニング後に Trax で予測モードに切り替える方法は?
ディープ ラーニング モデルを構築するための Trax ライブラリに慣れてきましたが、答えが見つからない 1 つの質問は、モデルのトレーニングが完了した後に "train" モードから "eval" モードに切り替える方法です。
トランスフォーマーを使用したこの例を検討してください ( https://github.com/google/trax#1-run-a-pre-trained-transformerから変更)
トレーニング後、モードを「予測」に切り替えるにはどうすればよいですか?
(同じ例に従って) 頭に浮かぶアイデアの 1 つは、モデルをトレーニングしてから保存し、モデルを再初期化することですが、今回は を使用してmode='train'、実行して重みをロードしますmodel.init_from_file(file)。
モデルを再初期化せずに直接行う方法はありますか?