CapsNet-pytorch

CapsNet-pytorch copied to clipboard

1. fix a mismatch between default and actual value of `lr` param. 2. some changes to default values for low memory. 3. compatibility change from `.data[0]` to `item()` for newer...

comsaint

In inference time , why use the label to reconstruct the image while not use the prediction of the model and the digit capsules as the input of reconstruction module?...

emobuAgent

Deprecated

2

/home/raul/Desktop/CapsNet-pytorch/net.py:28: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument. c = F.softmax(self.b) /home/raul/Desktop/CapsNet-pytorch/net.py:38: UserWarning: Implicit dimension choice for softmax has...

raul1968

I have a question: according to the paper, the squash function only be used after the sum of prediction u-hat? and in this code, there is a squash after the...

yimzhai3

After running the net.py file, the model file. pth was obtained. However, the following error reasons occurred while running the visualization file. How to solve them: Traceback (most recent call...

jisheng8