moved softmax to activations