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

fix attribute name for internal layer in DataParallel

This commit is contained in:
chenfeiyu 2021-01-11 16:56:55 +08:00
parent 39007e5bf8
commit ddfe2eda76
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/wavenet/train.py
View File

@ -62,7 +62,7 @@ class Experiment(ExperimentBase):
config.training.gradient_max_norm)) config.training.gradient_max_norm))
self.model = model self.model = model
self.model_core = model._layer if self.parallel else model self.model_core = model._layers if self.parallel else model
self.optimizer = optimizer self.optimizer = optimizer
def setup_dataloader(self): def setup_dataloader(self):