dmlc--dgl
VoVAllen
d0ea98befa
* add capsule example * clean code * better naming * better naming * Clean Codes based on pytorch MNIST example * Clean codes * Add README
447 B
447 B
DGL implementation of Capsule Network
This repo implements Hinton and his team's Capsule Network. Only margin loss is implemented, for simplicity to understand the DGL.
Training& Evaluation
# Run with default config
python main.py
# Run with train and test batch size 128, and for 50 epochs
python main.py --batch-size 128 --test-batch-size 128 --epochs 50