| 22 | } |
| 23 | |
| 24 | VARP squeezeNetExpr(int numClass) { |
| 25 | auto x = _Input({1, 3, 224, 224}, NC4HW4); |
| 26 | x = _Conv(0.0f, 0.0f, x, {3, 96}, {7, 7}, SAME, {2, 2}, {1, 1}, 1); |
| 27 | x = _MaxPool(x, {3, 3}, {2, 2}, SAME); |
| 28 | x = fireMoudle(x, 96, 16, 64, 64); |
| 29 | x = fireMoudle(x, 128, 16, 64, 64); |
| 30 | x = fireMoudle(x, 128, 32, 128, 128); |
| 31 | x = _MaxPool(x, {3, 3}, {2, 2}, SAME); |
| 32 | x = fireMoudle(x, 256, 32, 128, 128); |
| 33 | x = fireMoudle(x, 256, 48, 192, 192); |
| 34 | x = fireMoudle(x, 384, 48, 192, 192); |
| 35 | x = fireMoudle(x, 384, 64, 256, 256); |
| 36 | x = _MaxPool(x, {3, 3}, {2, 2}, SAME); |
| 37 | x = fireMoudle(x, 512, 64, 256, 256); |
| 38 | x = _Conv(0.0f, 0.0f, x, {512, numClass}, {1, 1}, VALID, {1, 1}, {1, 1}, 1); |
| 39 | x = _AvePool(x, {14, 14}, {1, 1}, VALID); |
| 40 | return x; |
| 41 | } |