Skip to content

Commit 8d26bb8

Browse files
committed
polish nn.Sequential attribute
1 parent b1f18f0 commit 8d26bb8

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

python/jittor/nn.py

+9
Original file line numberDiff line numberDiff line change
@@ -2308,6 +2308,15 @@ def __len__(self):
23082308

23092309
def named_children(self,):
23102310
return list(self.layers.items())
2311+
2312+
def __setattr__(self, key, value) -> None:
2313+
if isinstance(key, str) and key.isdigit():
2314+
if int(key)<len(self.layers):
2315+
self.add_module(key, value)
2316+
else:
2317+
super().__setattr__(key, value)
2318+
else:
2319+
super().__setattr__(key, value)
23112320

23122321

23132322
def __getattr__(self, key):

0 commit comments

Comments
 (0)