I tried this, but didn’t help. When I removed the F.relu from the last layer, the loss deviated a bit, but it went just ± 0.002, and the accuracy also didn’t improve.
I also added a large number of convolutional layers to try overfitting the model, but it still remains the same.