fix: reasonings init parameters are used now

2021-06-21 14:53:22 +02:00 · 2021-06-21 14:53:22 +02:00 · cfe09ec06b
commit cfe09ec06b
parent 3d76dffe3c
1 changed files with 3 additions and 1 deletions
--- a/prototorch/core/components.py
+++ b/prototorch/core/components.py
@ -253,8 +253,10 @@ class Reasonings(torch.nn.Module):
        self,
        distribution: Union[dict, list, tuple],
        initializer:
-        AbstractReasoningsInitializer = RandomReasoningsInitializer()):
+        AbstractReasoningsInitializer = RandomReasoningsInitializer(),
+    ):
        super().__init__()
+        self.add_reasonings(distribution, initializer)

    @property
    def num_classes(self):