0
nolearn.lasagne NeuralNetモデルに重みとバイアスを保存するにはどうすればよいですか?ドキュメントからは、NeuralNetの重みと偏りにアクセスしてそれらを保存する方法がわかりません。全体nolearnモデル(訓練歴、パラメータおよびアーキテクチャ)を保存するにはnolearn.lasagne NeuralNetモデルの保存方法と読み取り方法pickleを使用する
nolearn.lasagne NeuralNetモデルに重みとバイアスを保存するにはどうすればよいですか?ドキュメントからは、NeuralNetの重みと偏りにアクセスしてそれらを保存する方法がわかりません。全体nolearnモデル(訓練歴、パラメータおよびアーキテクチャ)を保存するにはnolearn.lasagne NeuralNetモデルの保存方法と読み取り方法pickleを使用する
、あなたがこれを行うことができますが:
import cPickle as pickle
sys.setrecursionlimit(10000) # you may need this if the network is large
with open("model_file", 'wb') as f:
pickle.dump(nolearnnet , f, -1)
包みあなたはGPU上でモデルを訓練し、上記を使用して、それを酸洗いだけにしたいので、予めご了承くださいそれをCPU上でアンピクル(またはその逆)すると、これは機能しません。その場合は、パラメータ値を保存するだけです。
weights = lasagne.layers.get_all_param_values(nolearnnet.get_all_layers()[-1])
これでこれらの重みを保存することができます。あなたが別のnolearnモデルにそれらをロードするときに、あなただけの次の操作を実行できます。
lasagne.layers.set_all_param_values(nolearnnet2.get_all_layers()[-1], weights)
それはこの議論を参照するために役立つことがあります。https://groups.google.com/forum/#!topic/lasagne-users/BbG95R6SZ0I