----------------------------------------------------------------
        Layer (type)               Output Shape         Param #
================================================================
            Conv2d-1          [-1, 8, 222, 222]             224
              ReLU-2          [-1, 8, 222, 222]               0
            Conv2d-3         [-1, 16, 220, 220]           1,168
              ReLU-4         [-1, 16, 220, 220]               0
            Conv2d-5         [-1, 32, 218, 218]           4,640
              ReLU-6         [-1, 32, 218, 218]               0
         MaxPool2d-7           [-1, 32, 72, 72]               0
            Conv2d-8           [-1, 64, 70, 70]          18,496
              ReLU-9           [-1, 64, 70, 70]               0
           Conv2d-10          [-1, 128, 68, 68]          73,856
          Flatten-11               [-1, 591872]               0
           Linear-12                  [-1, 128]      75,759,744
             ReLU-13                  [-1, 128]               0
           Linear-14                   [-1, 64]           8,256
             ReLU-15                   [-1, 64]               0
           Linear-16                   [-1, 32]           2,080
             ReLU-17                   [-1, 32]               0
           Linear-18                   [-1, 16]             528
             ReLU-19                   [-1, 16]               0
           Linear-20                    [-1, 8]             136
             ReLU-21                    [-1, 8]               0
           Linear-22                    [-1, 5]              45
          Softmax-23                    [-1, 5]               0
================================================================
Total params: 75,869,173
Trainable params: 75,869,173
Non-trainable params: 0
----------------------------------------------------------------
Input size (MB): 0.57
Forward/backward pass size (MB): 56.12
Params size (MB): 289.42
Estimated Total Size (MB): 346.12
----------------------------------------------------------------