Not a large MNIST model with 200 parameters, accuracy: 0.8353000283241272 loss: 0.5836790204048157
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━┓
┃ Layer (type) ┃ Output Shape ┃ Param # ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━┩
│ input_layer_8 (InputLayer) │ (None, 28, 28) │ 0 │
├──────────────────────────────────────┼─────────────────────────────┼─────────────────┤
│ reshape_16 (Reshape) │ (None, 28, 28, 1) │ 0 │
├──────────────────────────────────────┼─────────────────────────────┼─────────────────┤
│ depthwise_conv2d_32 │ (None, 13, 13, 5) │ 50 │
│ (DepthwiseConv2D) │ │ │
├──────────────────────────────────────┼─────────────────────────────┼─────────────────┤
│ depthwise_conv2d_33 │ (None, 6, 6, 5) │ 50 │
│ (DepthwiseConv2D) │ │ │
├──────────────────────────────────────┼─────────────────────────────┼─────────────────┤
│ depthwise_conv2d_34 │ (None, 2, 2, 5) │ 50 │
│ (DepthwiseConv2D) │ │ │
├──────────────────────────────────────┼─────────────────────────────┼─────────────────┤
│ depthwise_conv2d_35 │ (None, 1, 1, 10) │ 50 │
│ (DepthwiseConv2D) │ │ │
├──────────────────────────────────────┼─────────────────────────────┼─────────────────┤
│ reshape_17 (Reshape) │ (None, 10) │ 0 │
└──────────────────────────────────────┴─────────────────────────────┴─────────────────┘
Total params: 200 (800.00 B)
Trainable params: 200 (800.00 B)
Non-trainable params: 0 (0.00 B)