search

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

Network no longer trains correctly #18

Closed mathildor closed 6 years ago

mathildor commented 6 years ago

Both a colleague and I have tried to run your network, but it doesn't train and gets stuck on loss 0.36. Maybe one of the latest updates caused this?

Btw, thanks for sharing your code!

www0wwwjs1 commented 6 years ago

Could you tell me which script doesn't work and the specific parameters? I can check the code for bugs.

mathildor commented 6 years ago

I was running with mnist - python3 train.py "mnist". I saw that you did an update so I just started a new training, and seems to be working :)

www0wwwjs1 commented 6 years ago

Yes, I update the routing mechanism in the latest version. Everything should be ok this time😊.

发送自 Windows 10 版邮件https://go.microsoft.com/fwlink/?LinkId=550986应用

发件人: Mathilde Ørstavikmailto:notifications@github.com 发送时间: 2018年1月27日 3:15 收件人: www0wwwjs1/Matrix-Capsules-EM-Tensorflowmailto:Matrix-Capsules-EM-Tensorflow@noreply.github.com 抄送: Suofei Zhangmailto:zhangsuofei_1982@hotmail.com; Commentmailto:comment@noreply.github.com 主题: Re: [www0wwwjs1/Matrix-Capsules-EM-Tensorflow] Network no longer trains correctly (#18)

I was running with mnist - python3 train.py "mnist". I saw that you did an update so I just started a new training, and seems to be working :)

— You are receiving this because you commented. Reply to this email directly, view it on GitHubhttps://github.com/www0wwwjs1/Matrix-Capsules-EM-Tensorflow/issues/18#issuecomment-360877676, or mute the threadhttps://github.com/notifications/unsubscribe-auth/ANe0fpv_JlIN8W3han9W7BsJwAR5-f1Pks5tOiRAgaJpZM4Ri6K5.