Table 4.
Summary of each module in the presented model.
| Module | Kernel Size | Stride | Expansion Ratio | Parameters | Output Shape |
|---|---|---|---|---|---|
| Stem | 3 × 3 | 2 | − | 416 | (None, 112, 112, 32) |
| MBConv-A | 3 × 3 | 1 | 1 | 1544 | (None, 112, 112, 16) |
| MBConv-A | 3 × 3 | 2 | 6 | 6436 | (None, 56, 56, 24) |
| MBConv-B | 3 × 3 | 1 | 6 | 11,334 | (None, 56, 56, 24) |
| MBConv-A | 5 × 5 | 2 | 6 | 16,006 | (None, 28, 28, 40) |
| MBConv-B | 5 × 5 | 1 | 6 | 32,330 | (None, 28, 28, 40) |
| MBConv-A | 3 × 3 | 2 | 6 | 38,250 | (None, 14, 14, 80) |
| 2 × MBConv-B | 3 × 3 | 1 | 6 | 209,960 | (None, 14, 14, 80) |
| MBConv-A | 5 × 5 | 1 | 6 | 128,148 | (None, 14, 14, 112) |
| 2 × MBConv-B | 5 × 5 | 1 | 6 | 422,968 | (None, 14, 14, 112) |
| MBConv-A | 5 × 5 | 2 | 6 | 265,564 | (None, 7, 7, 192) |
| 3 × MBConv-B | 5 × 5 | 1 | 6 | 1,778,832 | (None, 7, 7, 192) |
| MBConv-A | 3 × 3 | 1 | 6 | 722,480 | (None, 7, 7, 320) |
| Head | 1 × 1 | 1 | − | 424,968 | (None, 8) |