Torch/Lua、訓練されたニューラルネットワークモデルをファイルに保存する方法は？

Question

私はTorch/Luaプロジェクトに取り組んでいました。このプロジェクトでは、人工ニューラルネットワークモデルを実装しました。 すべて動作しますが、今は次のようにコードを変更したいと思います。私の入力データセットは非常に大きいので、N = 20スパンで分割したいと思います。

次に、最初のデータセットのスパンでのみニューラルネットワークを訓練し、次に他のN-1 = 19スパンを並行してテストします。

これらの並列ジョブをすべて実行するには、ニューラルネットワークモデルの詳細をファイルに保存してから、19ジョブごとにロードする必要があります。

トーチが人工ニューラルネットワークモデルをファイルに正しく書き込む方法はありますか？

Answer 1

-- save the model 
torch.save(filename, model) 

。

-- load the model 
model = torch.load(filename)