def unfreezeParameters(net, single=True):
"""
Unfreeze parameters of a layer or a network so that they become trainable
again
Parameters
----------
net: a network layer
single: whether to freeze a single layer of all of the layers below as well
"""
all_layers = lasagne.layers.get_all_layers(net)
if single:
all_layers = [all_layers[-1]]
for layer in all_layers:
layer_params = layer.get_params()
for p in layer_params:
try:
layer.params[p].add('trainable')
except KeyError:
pass
model_helpers.py 文件源码
python
阅读 29
收藏 0
点赞 0
评论 0
评论列表
文章目录