Browse Source

fix(mge/utils): fix auto naming bug when expanding structure

GitOrigin-RevId: 5735bf5151
release-1.3
Megvii Engine Team 4 years ago
parent
commit
2369c4f607
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/python/megengine/module/sequential.py

+ 2
- 1
imperative/python/megengine/module/sequential.py View File

@@ -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

Loading…
Cancel
Save