model_helpers.py 文件源码

python
阅读 29 收藏 0 点赞 0 评论 0

项目:iterative_inference_segm 作者: adri-romsor 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号