C10H/CS286_BreakHis_CNN

CS286_BreakHis_CNN

Layer (type)	Output Shape	Param #
Conv2d-1	[-1, 32, 222, 222]	896
ReLU-2	[-1, 32, 222, 222]	0
MaxPool2d-3	[-1, 32, 111, 111]	0
Conv2d-4	[-1, 64, 111, 111]	18,496
ReLU-5	[-1, 64, 111, 111]	0
MaxPool2d-6	[-1, 64, 55, 55]	0
Conv2d-7	[-1, 128, 55, 55]	73,856
ReLU-8	[-1, 128, 55, 55]	0
MaxPool2d-9	[-1, 128, 27, 27]	0
Dropout-10	[-1, 128, 27, 27]	0
Flatten-11	[-1, 93312]	0
Linear-12	[-1, 128]	11,944,064
ReLU-13	[-1, 128]	0
Linear-14	[-1, 64]	8,256
ReLU-15	[-1, 64]	0
Linear-16	[-1, 1]	65
Total		12,045,633
Trainable params		12,045,633
Non-trainable params		0

Input size (MB): 0.57
Forward/backward pass size (MB): 48.63
Params size (MB): 45.95
Estimated Total Size (MB): 95.15