NeVerDieKinG

In file capsnet.py => function squash, ``` def squash(self, input_tensor): print(f'input_tensor.shape: {input_tensor.shape}') squared_norm = (input_tensor ** 2).sum(-1, keepdim=True) print(f'squared_norm.shape: {squared_norm.shape}') ``` Print out : ``` input_tensor.shape: torch.Size([100, 1, 10, 16,...

wrong use of classifier Layer for posterior_cls_prob

1

in file stacked_capsule_auto_encoder.py line 211 res.posterior_cls_prob = self.prior_classifier( should be res.posterior_cls_prob = self.posterior_classifier( otherwise it use same layer with prior_cls_prob