RVC-Boss 2024-03-13 17:03:59 +08:00 committed by GitHub
parent 931781774d
commit 7e012e7678
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -893,6 +893,7 @@ class SynthesizerTrn(nn.Module):
if freeze_quantizer: if freeze_quantizer:
self.ssl_proj.requires_grad_(False) self.ssl_proj.requires_grad_(False)
self.quantizer.requires_grad_(False) self.quantizer.requires_grad_(False)
self.quentizer.eval()
# self.enc_p.text_embedding.requires_grad_(False) # self.enc_p.text_embedding.requires_grad_(False)
# self.enc_p.encoder_text.requires_grad_(False) # self.enc_p.encoder_text.requires_grad_(False)
# self.enc_p.mrte.requires_grad_(False) # self.enc_p.mrte.requires_grad_(False)