Skip to content

Commit 4147940

Browse files
committed
fixed integer division error and multispeaker evaluation error
1 parent 897f31e commit 4147940

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

train.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -397,7 +397,7 @@ def eval_model(global_step, writer, device, model, checkpoint_dir, ismultispeake
397397
model_eval.load_state_dict(model.state_dict())
398398

399399
# hard coded
400-
speaker_ids = [0, 1, 10] if ismultispeaker else [None]
400+
speaker_ids = [0, 1, hparams.n_speakers-1] if ismultispeaker else [None]
401401
for speaker_id in speaker_ids:
402402
speaker_str = "multispeaker{}".format(speaker_id) if speaker_id is not None else "single"
403403

@@ -666,7 +666,7 @@ def train(device, model, data_loader, optimizer, writer,
666666
if hparams.masked_loss_weight > 0:
667667
# decoder output domain mask
668668
decoder_target_mask = sequence_mask(
669-
target_lengths / (r * downsample_step),
669+
target_lengths // (r * downsample_step),
670670
max_len=mel.size(1)).unsqueeze(-1)
671671
if downsample_step > 1:
672672
# spectrogram-domain mask

0 commit comments

Comments
 (0)