epoch 1, loss 0.0873, train acc 0.812, test acc 0.903