www0wwwjs1 / Matrix-Capsules-EM-Tensorflow

A Tensorflow implementation of CapsNet based on paper Matrix Capsules with EM Routing
Apache License 2.0
218 stars 83 forks source link

get a different loss and accuracy curve #31

Closed qimw closed 6 years ago

qimw commented 6 years ago

Hi, and I am using the default parameters training the net, but I got some different loss and training accuracy curves compared with the README file. Has anyone meet this kind of things?

yhyu13 commented 6 years ago

@qimw @www0wwwjs1

Your results looks pretty good.

I apologize for the confusion, the learning curves we proposed include weight decay actually. However, by default, the weight decay loss is set to false (here). From your result, we can tell that weight decay is a essential regularization.

qimw commented 6 years ago

Hi, thank you for your prompt reply : )