mirror of
https://github.com/wassname/TTS.git
synced 2026-09-11 12:00:24 +08:00
fix argument order for testing models
This commit is contained in:
@@ -38,7 +38,7 @@ class TacotronTrainTest(unittest.TestCase):
|
||||
|
||||
criterion = MSELossMasked().to(device)
|
||||
criterion_st = nn.BCEWithLogitsLoss().to(device)
|
||||
model = Tacotron2(24, c.r, 5).to(device)
|
||||
model = Tacotron2(num_chars=24, r=c.r, num_speakers=5).to(device)
|
||||
model.train()
|
||||
model_ref = copy.deepcopy(model)
|
||||
count = 0
|
||||
|
||||
@@ -45,8 +45,8 @@ class TacotronTrainTest(unittest.TestCase):
|
||||
criterion = L1LossMasked().to(device)
|
||||
criterion_st = nn.BCEWithLogitsLoss().to(device)
|
||||
model = Tacotron(
|
||||
32,
|
||||
5,
|
||||
num_chars=32,
|
||||
num_speakers=5,
|
||||
linear_dim=c.audio['num_freq'],
|
||||
mel_dim=c.audio['num_mels'],
|
||||
r=c.r,
|
||||
|
||||
Reference in New Issue
Block a user