Commit 4147940

committed

fixed integer division error and multispeaker evaluation error

1 parent 897f31e commit 4147940Copy full SHA for 4147940

File tree

-2

lines changed

-2

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -397,7 +397,7 @@ def eval_model(global_step, writer, device, model, checkpoint_dir, ismultispeake`
`397`	`397`	`model_eval.load_state_dict(model.state_dict())`
`398`	`398`
`399`	`399`	`# 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]`
`401`	`401`	`for speaker_id in speaker_ids:`
`402`	`402`	`speaker_str = "multispeaker{}".format(speaker_id) if speaker_id is not None else "single"`
`403`	`403`
`@@ -666,7 +666,7 @@ def train(device, model, data_loader, optimizer, writer,`
`666`	`666`	`if hparams.masked_loss_weight > 0:`
`667`	`667`	`# decoder output domain mask`
`668`	`668`	`decoder_target_mask = sequence_mask(`
`669`		`- target_lengths / (r * downsample_step),`
	`669`	`+ target_lengths // (r * downsample_step),`
`670`	`670`	`max_len=mel.size(1)).unsqueeze(-1)`
`671`	`671`	`if downsample_step > 1:`
`672`	`672`	`# spectrogram-domain mask`

Comments

(0)