mirror of
https://github.com/wassname/TTS.git
synced 2026-10-27 12:30:27 +08:00
bug fix
This commit is contained in:
1 parent
f4678cbd6b
commit
9d669d1024
1 file changed
+1
-1
@@ -44,7 +44,7 @@ class TacotronTrainTest(unittest.TestCase):
|
||||
stop_targets = (stop_targets.sum(2) >
|
||||
0.0).unsqueeze(2).float().squeeze()
|
||||
|
||||
criterion = L1LossMasked().to(device)
|
||||
criterion = L1LossMasked(seq_len_norm=False).to(device)
|
||||
criterion_st = nn.BCEWithLogitsLoss().to(device)
|
||||
model = Tacotron(
|
||||
num_chars=32,
|
||||
|
||||
Reference in new issue
Block a user