Closed ultranationalism closed 1 month ago
# Freeze vae and text_encoder and set unet to trainable def freeze_params(m: nn.Cell): for p in m.get_parameters(): p.require_grad = False
应为
def freeze_params(m: nn.Cell): for p in m.get_parameters(): p.requires_grad = False
fixed in #487
应为
Freeze vae and text_encoder and set unet to trainable