Table 1 parameter settings.
Parameters | Values |
|---|---|
Optimizer | Adam |
Loss | Mean-squared-error |
Batch-size | 256 |
Epochs | 10 |
Activation | ReLU |
Learning rate | 0.001 |
Kernel-size | 3*3 |
Dropout rate | 0.4 |
No. of channel in primary capsule layer | 32 |
Dimension of capsule in primary capsule layer | 8 |
Dimension of capsule in digit capsule layer | 16 |
No. of routing iterations | 2 |
