def load_model(net, layer='fc8'): model_values = utils.PickleLoad(os.path.join(model_dir, 'caffe_reference_%s.pkl' % layer)) lasagne.layers.set_all_param_values(net[layer], model_values)