Moore-AnimateAnyone icon indicating copy to clipboard operation
Moore-AnimateAnyone copied to clipboard

Some top layer parames of reference_unet don't need grad, what is the reason?

Open BugsMaker0513 opened this issue 10 months ago • 1 comments

Some top layer parames of reference_unet don't need grad

for name, param in reference_unet.named_parameters():
    if "up_blocks.3" in name:
        param.requires_grad_(False)
    else:
        param.requires_grad_(True)

BugsMaker0513 avatar Apr 22 '24 07:04 BugsMaker0513