dmlc--dgl
VoVAllen
d0ea98befa
* add capsule example * clean code * better naming * better naming * Clean Codes based on pytorch MNIST example * Clean codes * Add README
13 行
447 B
Markdown
13 行
447 B
Markdown
DGL implementation of Capsule Network
|
|
=====================================
|
|
|
|
This repo implements Hinton and his team's [Capsule Network](https://arxiv.org/abs/1710.09829).
|
|
Only margin loss is implemented, for simplicity to understand the DGL.
|
|
|
|
## Training& Evaluation
|
|
```bash
|
|
# 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
|
|
``` |