add standalone module

keras-team · Jul 27, 2024 · d0a17fa · d0a17fa
1 parent 66126fe
commit d0a17fa
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/keras/src/backend/torch/layer.py b/keras/src/backend/torch/layer.py
@@ -140,10 +140,8 @@ def _setattr_hook(self, name, value):
             and all(isinstance(v, Layer) for v in value)
             and len(value) > 0
         ):
-            warnings.warn(
-                "Torch backend currently cannot track list of "
-                "layers properly as an attribute."
-            )
+            for idx, v in enumerate(value):
+                self.add_module(f"{name}_{idx}", v)
 
         return name, value