From cfe09ec06b92336e9941b0f5e18a965b6bafb871 Mon Sep 17 00:00:00 2001 From: Alexander Engelsberger Date: Mon, 21 Jun 2021 14:53:22 +0200 Subject: [PATCH] fix: reasonings init parameters are used now --- prototorch/core/components.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/prototorch/core/components.py b/prototorch/core/components.py index e9c6433..5d474a7 100644 --- 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):