enable weights for softmax cross entropy loss layer