In the above program, I have used:=
batch size = 20, i.e torch.size[20, 24] last one is [19,24]
in_features = 480
out_features = 240
out_classes = 2 
I have kept Kernel_size = 5, 

Being a beginner, not sure what to set input and output parameters in nn.linear() modules.
I have removed the pooling layers. Everytime I make a change, getting the error
RuntimeError: mat1 and mat2 shapes cannot be multiplied (1x3072 and 7680x240)