Capsule network | Deep capsule network | |
Convolution layer | Conv1: 256*9*9 | Conv1: 512*9*9 Conv2: 256*5*5 |
Primary Capsule | 9*9 | 5*5 |
Digit Capsule | One time dynamic routing | Twice dynamic routing |
FC |
Operating system | Windows10(RAM16.0GB) |
CPU | Intel(R)Core(TM)i7-9750H |
GPU | NVIDIA GeForce GTX 1660 Ti |
Dataset | MNIST |
Other | pytorch1.5.0+cu101 python 3.7.7 |