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

Merge pull request #82 from iclementine/develop 

fix: the condition to init DataParallel
This commit is contained in:
Feiyu Chan 2021-01-11 17:19:30 +08:00 committed by GitHub
parent c615de2354 e53b9a0745
commit ede6835bd2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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, n_mels=config.data.n_mels,
kernel_size=config.model.kernel_size) kernel_size=config.model.kernel_size)
if self.parallel > 1: if self.parallel:
model = paddle.DataParallel(model) model = paddle.DataParallel(model)
optimizer = paddle.optimizer.Adam( optimizer = paddle.optimizer.Adam(
config.training.lr, parameters=model.parameters()) 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, loss_type=config.model.loss_type,
log_scale_min=config.model.log_scale_min) log_scale_min=config.model.log_scale_min)
if self.parallel > 1: if self.parallel:
model = paddle.DataParallel(model) model = paddle.DataParallel(model)
lr_scheduler = paddle.optimizer.lr.StepDecay( lr_scheduler = paddle.optimizer.lr.StepDecay(