You already have the answer to your question, but I just want to point that you are using ReLU in the last layer, but you said you were doing classification. I think you should use SoftMax there, no ReLU.