设置不同的学习率

1
y.update([x])

用字典x更新y字典,若字典x中的键与字典y中的键相同则更新,否则在字典y中追加该键。具体代码如下所示,注:sgd返回的是一个字典。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
learning_rate = 0.01				
Layers=lasagne.layers.get_all_layers(network)
updates = collections.OrderedDict()
updates.update(lasagne.updates.sgd(
loss,
Layers[-1].get_params(trainable=True),
learning_rate*0.1
))
for i in range(len(Layers)-1):
updates.update(lasagne.updates.sgd(
loss,
Layers[i].get_params(trainable=True),
learning_rate
))