Initialize modules

experimaestro · Nov 29, 2023 · de68d02 · de68d02
1 parent 4ba979b
commit de68d02
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/xpmir/learning/optim.py b/src/xpmir/learning/optim.py
@@ -91,6 +91,10 @@ def __init__(self):
         Initializable.__init__(self)
         torch.nn.Module.__init__(self)
 
+    def __initialize__(self, *args, **kwargs):
+        for module in self.modules:
+            module.initialize(*args, **kwargs)
+
     def __call__(self, *args, **kwargs):
         raise AssertionError("This module cannot be used as such: it is just a ")