forgetest3
/
ParakeetParakeet
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix argument order

This commit is contained in:
chenfeiyu 2021-04-22 13:46:51 +08:00
parent 3a744dbf30
commit c2560e8aa2
2 changed files with 4 additions and 5 deletions
examples/tacotron2_aishell3
config.pytrain.py

1
examples/tacotron2_aishell3/config.py
View File

@ -60,7 +60,6 @@ _C.training = CN(
lr=1e-3, # learning rate
weight_decay=1e-6, # the coeff of weight decay
grad_clip_thresh=1.0, # the clip norm of grad clip.
plot_interval=1000, # plot attention and spectrogram
valid_interval=1000, # validation
save_interval=1000, # checkpoint
max_iteration=500000, # max iteration to train

8
examples/tacotron2_aishell3/train.py
View File

@ -54,11 +54,11 @@ class Experiment(ExperimentBase):
self.model.train()
texts, tones, mels, utterance_embeds, text_lens, output_lens, stop_tokens = batch
outputs = self.model(texts,
mels,
text_lens,
mels,
output_lens,
tones=tones,
utterance_embeds=utterance_embeds)
global_condition=utterance_embeds)
losses = self.compute_losses(batch, outputs)
loss = losses["loss"]
loss.backward()
@ -87,11 +87,11 @@ class Experiment(ExperimentBase):
for i, batch in enumerate(self.valid_loader):
texts, tones, mels, utterance_embeds, text_lens, output_lens, stop_tokens = batch
outputs = self.model(texts,
mels,
text_lens,
mels,
output_lens,
tones=tones,
utterance_embeds=utterance_embeds)
global_condition=utterance_embeds)
losses = self.compute_losses(batch, outputs)
for key, value in losses.items():
valid_losses[key].append(float(value))