Function apply_wd

k_diffusion/models/modules.py:77–81

(module)

def apply_wd(module):
    for name, param in module.named_parameters():
        if name.endswith("weight"):
            tag_param(param, "wd")
    return module


def filter_params(function, module):