From af62cf6f051255a9f9b9789b2c917ac13ae5f10e Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 9 Apr 2021 18:00:16 +0800 Subject: [PATCH] fix(mge/utils): fix auto naming bug when expanding structure GitOrigin-RevId: 5735bf51513a47b64bf5cdf6a065d3cdf1a8de56 --- imperative/python/megengine/module/sequential.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/imperative/python/megengine/module/sequential.py b/imperative/python/megengine/module/sequential.py index b4dbdafd2..9953140a6 100644 --- a/imperative/python/megengine/module/sequential.py +++ b/imperative/python/megengine/module/sequential.py @@ -92,6 +92,7 @@ class Sequential(Module): return [getattr(self, key) for key in self.layer_keys] def forward(self, inp): - for layer in self.layer_values: + # avoid layer_values as a name prefix, see Module.__getattribute__ + for layer in [getattr(self, key) for key in self.layer_keys]: inp = layer(inp) return inp -- GitLab