CapsNet-Pytorch icon indicating copy to clipboard operation
CapsNet-Pytorch copied to clipboard

squeeze function

Open qiaokang0 opened this issue 8 years ago • 3 comments

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

qiaokang0 avatar Jan 16 '18 02:01 qiaokang0

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'

dragen1860 avatar Jan 17 '18 07:01 dragen1860

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

qiaokang0 avatar Jan 18 '18 07:01 qiaokang0

ok, thank you.

dragen1860 avatar Jan 18 '18 08:01 dragen1860