Merge branch 'develop' into release/v0.2

2021-01-11 17:00:35 +08:00 · 2021-01-11 17:00:35 +08:00 · 289827c9cf
parent 610b8f2cef c615de2354
commit 289827c9cf
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):