Function zero_module

demo_utils/utils.py:196–199

(module)

@torch.no_grad()
def zero_module(module):
    for p in module.parameters():
        p.detach().zero_()
    return module


@torch.no_grad()