pip install torchinfo
			
			

			
    from torchinfo import summary

    summary(mymodel)						
			
			

			
=================================================================
Layer (type:depth-idx)                   Param #
=================================================================
mymodel                                  --
├─Sequential: 1-1                        --
│    └─Conv2d: 2-1                       640
│    └─ReLU: 2-2                         --
│    └─MaxPool2d: 2-3                    --
├─Sequential: 1-2                        --
│    └─Conv2d: 2-4                       73,856
│    └─ReLU: 2-5                         --
│    └─MaxPool2d: 2-6                    --
├─Sequential: 1-3                        --
│    └─Conv2d: 2-7                       295,168
│    └─ReLU: 2-8                         --
│    └─MaxPool2d: 2-9                    --
├─Sequential: 1-4                        --
│    └─Conv2d: 2-10                      1,180,160
│    └─ReLU: 2-11                        --
│    └─MaxPool2d: 2-12                   --
├─Sequential: 1-5                        --
│    └─Flatten: 2-13                     --
│    └─Linear: 2-14                      62,918,656
│    └─Dropout: 2-15                     --
│    └─ReLU: 2-16                        --
│    └─Linear: 2-17                      589,968
=================================================================
Total params: 65,058,448
Trainable params: 65,058,448
Non-trainable params: 0
=================================================================