YuxianMeng / Matrix-Capsules-pytorch

A Pytorch implementation of "Matrix Capsules with EM routing"
234 stars 48 forks source link

Do you succeed in reproducing the author's result on smallNORB? #1

Open yhyu13 opened 6 years ago

yhyu13 commented 6 years ago

Hi, @shzygmyx

I am a contributor to another matrix CapsNet in TensorFlow. Are you able to reproduce the result suggested by the author of the paper?

YuxianMeng commented 6 years ago

I haven't try training CapsNet on SmallNORB since I'm not familiar with this dataset and I'm busy with my application for graduate school. I'd appreciate it if you can help test my program on smallNORB :)

ZHANG-SHI-CHANG commented 6 years ago

when i try training CapsNet on SmallNORB, it is only 20% accuracy with 1 iter, i don't know where is wrong , i will try more iter, :)

YuxianMeng commented 6 years ago

@ZHANG-SHI-CHANG Thanks for your contribution :) I'm not quite sure about this, but maybe you can try to use smaller learning rate and change the lambda/m schedule

OlivierX commented 5 years ago

Same issue here...