From ddfe2eda7635427bf8e3256ac5b9a6422eecac51 Mon Sep 17 00:00:00 2001
From: chenfeiyu <chenfeiyu@baidu.com>
Date: Mon, 11 Jan 2021 16:56:55 +0800
Subject: [PATCH] fix attribute name for internal layer in DataParallel

---
 examples/wavenet/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/wavenet/train.py b/examples/wavenet/train.py
index 8e9bc0e..166e23d 100644
--- 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):