Added weights to softmax loss layer.