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

fix: the condition to init DataParallel

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

2
examples/waveflow/train.py
View File

@ -46,7 +46,7 @@ class Experiment(ExperimentBase):
n_mels=config.data.n_mels,
kernel_size=config.model.kernel_size)
if self.parallel > 1:
if self.parallel:
model = paddle.DataParallel(model)
optimizer = paddle.optimizer.Adam(
config.training.lr, parameters=model.parameters())

2
examples/wavenet/train.py
View File

@ -49,7 +49,7 @@ class Experiment(ExperimentBase):
loss_type=config.model.loss_type,
log_scale_min=config.model.log_scale_min)
if self.parallel > 1:
if self.parallel:
model = paddle.DataParallel(model)
lr_scheduler = paddle.optimizer.lr.StepDecay(