Bo Liu

icon company The University of Texas at Austin icon location Be a scientist out of interest :)

Results 2 issues of Bo Liu

The model is not improving.

2 comment

Hi, after I cloned your repo and ran the dagger.py, the model doesn't seem to improve. In fact, it stops at 28 iterations every time. I didn't change any parameter...

Why dim=2 here? I guess the softmax is taken over all capsules in the layer above, so shouldn't that be dim=0?

4 comment

https://github.com/gram-ai/capsule-networks/blob/1a4edd27a0ed73232cb266c85091d712854f3e71/capsule_network.py#L69