Function zero_module

lib_layerdiffuse/vae.py:19–25

Zero out the parameters of a module and return it.

(module)

def zero_module(module):
    """
    Zero out the parameters of a module and return it.
    """
    for p in module.parameters():
        p.detach().zero_()
    return module


class LatentTransparencyOffsetEncoder(torch.nn.Module):