Branch: master
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
DGLDigitCapsule.py
DGLRoutingLayer.py
README.md
main.py
model.py
simple_routing.py

README.md

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.

Dependencies

  • PyTorch 0.4.1+
  • torchvision
pip install torch torchvision

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