def freezeParameters(net, single=True):
"""
Freeze parameters of a layer or a network so that they are not trainable
anymore
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].remove('trainable')
except KeyError:
pass
model_helpers.py 文件源码
python
阅读 28
收藏 0
点赞 0
评论 0
评论列表
文章目录