CapsNet-Pytorch (base_capsNet)

A Pytorch implementation of CapsNet based on Geoffrey Hinton's paper Dynamic Routing Between Capsules

Requirements

torch==1.8.1

torchvision==0.9.1

numpy==1.19.2

run

python main.py