lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

polish nn.Sequential attribute

This commit is contained in:
lidongyang 2024-06-05 22:31:20 +08:00
parent c49be7cf79
commit 393684f196
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
python/jittor/nn.py
View File

@ -2308,6 +2308,15 @@ class Sequential(Module):
def named_children(self,): def named_children(self,):
return list(self.layers.items()) return list(self.layers.items())
def __setattr__(self, key, value) -> None:
if isinstance(key, str) and key.isdigit():
if int(key)<len(self.layers):
self.add_module(key, value)
else:
super().__setattr__(key, value)
else:
super().__setattr__(key, value)
def __getattr__(self, key): def __getattr__(self, key):