higgsfield / Capsule-Network-Tutorial

Pytorch easy-to-follow Capsule Network tutorial
767 stars 135 forks source link

Fix initialization parameters in PrimaryCaps class #14

Closed donglinz closed 3 years ago

donglinz commented 5 years ago

Fix issue 13 https://github.com/higgsfield/Capsule-Network-Tutorial/issues/13