Как хранить и читать nolearn.lasagne Модели NeuralNet с использованием pickle

Question

Как хранить вес и смещения в модели nolearn.lasagne NeuralNet? Из документации я не вижу, как получить доступ к весам и предубеждениям NeuralNet и сохранить их.

Answer 1

Чтобы сохранить всю модель nolearn (история обучения, параметры и архитектуры), вы можете сделать это:

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