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

Merge pull request #80 from iclementine/develop 

wavenet: fix attribute name for internal layer in DataParallel
This commit is contained in:
Feiyu Chan 2021-01-11 16:58:34 +08:00 committed by GitHub
parent 39007e5bf8 ddfe2eda76
commit c615de2354
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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))
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
def setup_dataloader(self):