From 393684f196cde5252cecdd6809ad945135c1d91a Mon Sep 17 00:00:00 2001
From: lidongyang <lidongyang8029@163.com>
Date: Wed, 5 Jun 2024 22:31:20 +0800
Subject: [PATCH] polish nn.Sequential attribute

---
 python/jittor/nn.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/python/jittor/nn.py b/python/jittor/nn.py
index 8f8012a8..b423b45c 100644
--- a/python/jittor/nn.py
+++ b/python/jittor/nn.py
@@ -2308,6 +2308,15 @@ class Sequential(Module):
     
     def named_children(self,):
         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):