CapsNet-Pytorch
CapsNet-Pytorch copied to clipboard
squeeze function
hi,i think you mix up mod and mod_sq in the squeeze function, it would be great that you check if i am right
emmm, currently I have not check your problem. I remember the formula is listed in original paper. Please tell me the performance difference when you correct the 'mistake'
in line 111: return (mod / (1 + mod)) * (input / mod_sq) while the formula in paper should be return (mod_sq / (1 + mod_sq)) * (input / mod) both of these two 'ways' can lead good correct rates
ok, thank you.