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

Using Batch Norm & Using numeric check to cope with NaN without interrupt training #11

Closed yhyu13 closed 6 years ago

yhyu13 commented 6 years ago

Using batch norm layer to achieve normalized inputs. Change batch size to 50. Using reduce_mean to judge loss independently from batch size.

yhyu13 commented 6 years ago

Using numeric check is effective discard_nan

You can now review real time tensorboard (because it used to be confused by models checkpoints) realtime