7cda122d9636cdfe8dab5c745b3fa4d30ab57cc9,gluon/utils.py,,prepare_model,#Any#Any#Any#Any#Any#Any#,171

Before Change

        net.initialize(mx.init.MSRAPrelu(), ctx=ctx)

    if tune_layers:
        tune_layers_ptrn = tuple(tune_layers.split(","))
        params = net._collect_params_with_prefix()
        param_keys = list(params.keys())
        for key in param_keys:
            if not key.startswith(tune_layers_ptrn):

After Change

    if tune_layers:
        tune_layers_pattern = re.compile(tune_layers)
        for k, v in net._collect_params_with_prefix().items():
            if tune_layers_pattern.match(k):
                logging.info("Fine-tune parameter: {}".format(k))
            else:
                v.grad_req = "null"
        for param in net.collect_params().values():
            if param._data is not None:
                continue
            param.initialize(mx.init.MSRAPrelu(), ctx=ctx)

Instances