Layer (type)               Output Shape         Param #

================================================================
Conv2d-1        [128, 32, 250, 250]             896
ReLU-2        [128, 32, 250, 250]               0
MaxPool2d-3        [128, 32, 125, 125]               0
Dropout-4        [128, 32, 125, 125]               0
Conv2d-5        [128, 64, 125, 125]          18,496
ReLU-6        [128, 64, 125, 125]               0
MaxPool2d-7          [128, 64, 62, 62]               0
Dropout-8          [128, 64, 62, 62]               0
Conv2d-9         [128, 128, 62, 62]          73,856
ReLU-10         [128, 128, 62, 62]               0
MaxPool2d-11         [128, 128, 32, 32]               0
Dropout-12         [128, 128, 32, 32]               0
Linear-13                 [128, 240]      31,457,520
Linear-14                 [128, 240]      31,457,520
Linear-15                   [128, 2]             482
Total params: 63,008,770
Trainable params: 63,008,770
Non-trainable params: 0
Input size (MB): 91.55
Forward/backward pass size (MB): 8533.91
Params size (MB): 240.36
Estimated Total Size (MB): 8865.82