Capsule-Network-Tutorial

Capsule-Network-Tutorial copied to clipboard

hello,I copy the code in the pycharm,but I found that the weight are 'nan',why?

Sample-design-alt

Similar to others, I found many issues with this implementation (a lot of mistakes!!!). So I decided to create my own one. It is bug-free and works very good. You...

amobiny

Wrong stack when cat the capsule

2

 Stack should not be implemeted on dim 1，but dim 4. Otherwise, the view result will be wrong.

BoPang1996

To handle > IndexError:_ invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

ketankr9

I don't know why the train accuracy is becoming lower and lower with the epoch decreasing. Besides, the margin loss and reconstruction loss can't be caculated.

yimzhai3

Softmax in routing algorithm incorrect?

1

Hi, I think the softmax in the routing algorithm is being calculated over the wrong dimension. Currently the code has: ``` b_ij = Variable(torch.zeros(1, self.num_routes, self.num_capsules, 1)) ... for iteration...

geefer

In the latest version of Pytorch loss.data returns the value, loss.data[0] throws an error.

kunalmessi10

margin loss wrong

2

Hello, I think your magin loss is wrong, according to the paper, after relu, there is a square you forgot

DanlanChen

RuntimeError:index_select() and issue about DataParallel

1

First of all, thanks, its definitely an easy to follow CapsNet tutorial for me as a beginner, but I found an error after running the code: RuntimeError: index_select(): argument 'index'...

ChengHuang-CH

I think **index=.....** should be wrapped in as **index=Variable(.....)** in Decoder class as: masked = masked.index_select(dim=0, index=Variable(max_length_indices.squeeze(1).data))

shirishr