testadmin
/
ParakeetEricRoss
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix a bug when using a method other than forward with DataParallel

This commit is contained in:
chenfeiyu 2021-01-11 17:24:46 +08:00
parent e53b9a0745
commit 7c5e98dfb3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/wavenet/train.py
View File

@ -119,7 +119,7 @@ class Experiment(ExperimentBase):
mel, wav, audio_starts = batch
y = self.model(wav, mel, audio_starts)
loss = self.model.loss(y, wav)
loss = self.model_core.loss(y, wav)
loss.backward()
self.optimizer.step()
iteration_time = time.time() - start
@ -141,7 +141,7 @@ class Experiment(ExperimentBase):
valid_losses = []
mel, wav, audio_starts = next(valid_iterator)
y = self.model(wav, mel, audio_starts)
loss = self.model.loss(y, wav)
loss = self.model_core.loss(y, wav)
valid_losses.append(float(loss))
valid_loss = np.mean(valid_losses)
self.visualizer.add_scalar(