| Layer (type) | Output shape | Params |
| Conv2d-1 | [−1, 16, 224, 224] | 464 | DPM-2 | [−1, 32, 112, 112] | 1968 | DPM-3 | [−1, 48, 56, 56] | 3888 | DPM-4 | [−1, 64, 28, 28] | 6720 | DPM-5 | [−1, 80, 14, 14] | 10320 | DPM-6 | [−1, 96, 7, 7] | 14688 | DW-7 | [−1, 112, 7, 7] | 12032 | DDS-8 | [−1, 16, 7, 7] | 3056 | DDS-9 | [−1, 16, 7, 7] | 3488 | DDS-10 | [−1, 16, 7, 7] | 3920 | DDS-11 | [−1, 16, 7, 7] | 4352 | DDS-12 | [−1, 16, 7, 7] | 4784 | DDS-13 | [−1, 16, 7, 7] | 5216 | DDS-14 | [−1, 16, 7, 7] | 5648 | DDS-15 | [−1, 16, 7, 7] | 6080 | DDS-16 | [−1, 16, 7, 7] | 6512 | Conv2d-17 | [−1, 480, 1, 1] | 124320 | SoftMax-18 | SoftMax-18 | 1443 | Total params | 218755 | | Input size (MB) | 0.57 | |
|
|