From 8c4316cd5e1036575a0689d400a0a06e9e745aa7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Wed, 6 Sep 2023 11:39:25 +0800 Subject: [PATCH] del freeze enc --- train_ms.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/train_ms.py b/train_ms.py index bc72253..325bee9 100644 --- a/train_ms.py +++ b/train_ms.py @@ -114,12 +114,6 @@ def run(): noise_scale_delta = noise_scale_delta, **hps.model).cuda(rank) - freeze_enc = getattr(hps.model, "freeze_enc", False) - if freeze_enc: - print("freeze encoder !!!") - for param in net_g.enc_p.parameters(): - param.requires_grad = False - net_d = MultiPeriodDiscriminator(hps.model.use_spectral_norm).cuda(rank) optim_g = torch.optim.AdamW( filter(lambda p: p.requires_grad, net_g.parameters()),