2016-12-20 1 views

答えて

0

、あなたがこれを行うことができますが:

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

関連する問題