julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: reasonings init parameters are used now

Browse Source
This commit is contained in:
Alexander Engelsberger 2021-06-21 14:53:22 +02:00
parent 3d76dffe3c
commit cfe09ec06b
No known key found for this signature in database
GPG Key ID: BE3F5909FF0D83E3
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
prototorch/core/components.py
View File

@ -253,8 +253,10 @@ class Reasonings(torch.nn.Module):
self, self,
distribution: Union[dict, list, tuple], distribution: Union[dict, list, tuple],
initializer: initializer:
AbstractReasoningsInitializer = RandomReasoningsInitializer()): AbstractReasoningsInitializer = RandomReasoningsInitializer(),
):
super().__init__() super().__init__()
self.add_reasonings(distribution, initializer)
@property @property
def num_classes(self): def num_classes(self):