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):