paddle.static.sparsity. reset_excluded_layers ( main_program=None ) [source]

Reset exculded layers setting corresponding to main_program. If main_program is None, then all configurations of excluded_layers would be cleaned.

  • main_program (Program, optional) – Program with model definition and its parameters.

  • Examples

  • code-block: (.) –

    python: import paddle from paddle.static import sparsity


    main_program = paddle.static.Program() startup_program = paddle.static.Program()

    with paddle.static.program_guard(main_program, startup_program):

    input_data =’data’, shape=[None, 128]) label =’label’, shape=[None, 10]) hidden = paddle.static.nn.fc(x=input_data, num_flatten_dims=-1, size=32, activation=None, name=”my_first_fc”) hidden = paddle.static.nn.fc(x=hidden, num_flatten_dims=-1, size=32, activation=None, name=”my_second_fc”) prob = paddle.static.nn.fc(x=hidden, num_flatten_dims=-1, size=10, activation=None) loss = paddle.mean(paddle.nn.functional.square_error_cost(prob, label))

    # Setup exluded layers out from ASP workflow. # Please note, excluded_layers must be set before calling optimizer.minimize(). sparsity.set_excluded_layers(main_program, [“my_second_fc”]) # Now the weights of “my_second_fc” would not be included in Automatic SParsity’s workflow.

    # Reset excluded_layers, all FC layers would be included into Automatic SParsity’s workflow. # Please note, reset_excluded_layers also must be called before calling optimizer.minimize(). sparsity.reset_excluded_layers(main_program)