weighted softmax loss