fix attribute name for internal layer in DataParallel

2021-01-11 16:56:55 +08:00 · 2021-01-11 16:56:55 +08:00 · ddfe2eda76
parent 39007e5bf8
commit ddfe2eda76
1 changed files with 1 additions and 1 deletions
--- a/examples/wavenet/train.py
+++ b/examples/wavenet/train.py
@ -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):