tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

del freeze enc

Browse Source
This commit is contained in:
Stardust·减
2023-09-06 11:39:25 +08:00
committed by GitHub
parent 371ffeb98e
commit 8c4316cd5e
1 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
train_ms.py
View File

@@ -114,12 +114,6 @@ def run():
noise_scale_delta = noise_scale_delta, noise_scale_delta = noise_scale_delta,
**hps.model).cuda(rank) **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) net_d = MultiPeriodDiscriminator(hps.model.use_spectral_norm).cuda(rank)
optim_g = torch.optim.AdamW( optim_g = torch.optim.AdamW(
filter(lambda p: p.requires_grad, net_g.parameters()), filter(lambda p: p.requires_grad, net_g.parameters()),