PyTorch / TensorFlow / Kerasで開発されたトレーニング済みニューラルネットワークを保存して使用するにはどうすればよいですか?

neo4k

モデルだけをトレーニングして他の人と共有した後、モデルを保存する方法はありますか?通常のスクリプトのように?ネットワークはフロート行列のコレクションであるため、ユーザーにこれらのフレームワークのインストールを要求する代わりに、これらのトレーニングされた重みを抽出し、新しいデータで実行して予測を行うことは可能ですか?私はこれらのフレームワークに不慣れであり、必要に応じて説明を行います。

ランダムウォーカー

PyTorch:この投稿説明されているように、モデルのパラメーターをディクショナリとして保存するか、ディクショナリをロードしてモデルのパラメーターを設定できます。PyTorchモデルをオブジェクトとして保存/ロードすることもできます。どちらの手順でも、効率的な行列乗算などのために、ユーザーは少なくとも1つのテンソル計算フレームワークをインストールする必要があります。

この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。

侵害の場合は、連絡してください[email protected]

編集
0

コメントを追加

0

関連記事

Related 関連記事

ホットタグ

アーカイブ