Branch: master
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

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.


  • PyTorch 0.4.1+
  • torchvision
pip install torch torchvision

Training & Evaluation

# Run with default config
# Run with train and test batch size 128, and for 50 epochs
python --batch-size 128 --test-batch-size 128 --epochs 50