dmlc--dgl
Giuseppe Futia
6c3dba867a
* Update prerequisites of README * dependencies for pytorch models * dependencies for mxnet models * minor
24 行
572 B
Markdown
24 行
572 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.
|
|
|
|
Dependencies
|
|
--------------
|
|
* PyTorch 0.4.1+
|
|
* torchvision
|
|
|
|
```bash
|
|
pip install torch torchvision
|
|
```
|
|
|
|
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
|
|
```
|