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

Yhyu13: refactory duplicate code #8

Closed yhyu13 closed 6 years ago

yhyu13 commented 6 years ago

To train:

python train.py "mnist" python train.py "smallNORB"

To evaluate:

python eval.py "mnist" python eval.py "smallNORB"