ptrblck:

only contain conv and linear layers


My model has only linear and conv layers. Batch norm layers are not there.
Also I think the particular lr, bs etc configuration was the reason for low accuracy as I observed that in the next run with other lr value etc. the network was again showing expected performance.
Is that expected to happen?