Update model

xcmyz · Jul 20, 2020 · 0b01f67 · 0b01f67
1 parent 20c9d8e
commit 0b01f67
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/model.py b/model.py
@@ -21,7 +21,7 @@ def __init__(self):
         self.length_regulator = LengthRegulator()
         self.decoder = Decoder()
 
-        self.mel_linear = Linear(hp.decoder_output_size, hp.num_mels)
+        self.mel_linear = Linear(hp.decoder_dim, hp.num_mels)
         self.postnet = CBHG(hp.num_mels, K=8,
                             projections=[256, hp.num_mels])
         self.last_linear = Linear(hp.num_mels * 2, hp.num_mels)

diff --git a/modules.py b/modules.py
@@ -97,7 +97,7 @@ class DurationPredictor(nn.Module):
     def __init__(self):
         super(DurationPredictor, self).__init__()
 
-        self.input_size = hp.d_model
+        self.input_size = hp.encoder_dim
         self.filter_size = hp.duration_predictor_filter_size
         self.kernel = hp.duration_predictor_kernel_size
         self.conv_output_size = hp.duration_predictor_filter_size

diff --git a/train.py b/train.py
@@ -37,7 +37,7 @@ def main(args):
                                  betas=(0.9, 0.98),
                                  eps=1e-9)
     scheduled_optim = ScheduledOptim(optimizer,
-                                     hp.d_model,
+                                     hp.decoder_dim,
                                      hp.n_warm_up_step,
                                      args.restore_step)
     fastspeech_loss = DNNLoss().to(device)