Function clear_act_buffer

lib/model_wrapper_low.py:122–125

(act_model)

def clear_act_buffer(act_model):
    for name, module in act_model.named_modules():
        if isinstance(module, ActLinear):
            module.clear_act_buffer()


def make_low_rank(