Skip to content

Commit

Permalink
Merge pull request #214 from Misterion777/master
Browse files Browse the repository at this point in the history
Integer division error and multispeaker evaluation error
  • Loading branch information
r9y9 authored Oct 23, 2020
2 parents 897f31e + 4147940 commit 3b549ec
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -397,7 +397,7 @@ def eval_model(global_step, writer, device, model, checkpoint_dir, ismultispeake
model_eval.load_state_dict(model.state_dict())

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

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

0 comments on commit 3b549ec

Please sign in to comment.