(self, spec_channels, z_latent_dim, emb_dim)
| 765 | |
| 766 | class MelStyleEncoderVAE(nn.Module): |
| 767 | def __init__(self, spec_channels, z_latent_dim, emb_dim): |
| 768 | super().__init__() |
| 769 | self.ref_encoder = MelStyleEncoder(spec_channels, style_vector_dim=emb_dim) |
| 770 | self.fc1 = nn.Linear(emb_dim, z_latent_dim) |
| 771 | self.fc2 = nn.Linear(emb_dim, z_latent_dim) |
| 772 | self.fc3 = nn.Linear(z_latent_dim, emb_dim) |
| 773 | self.z_latent_dim = z_latent_dim |
| 774 | |
| 775 | def reparameterize(self, mu, logvar): |
| 776 | if self.training: |
nothing calls this directly
no test coverage detected