Fix bugs in WeakLabelLossLayer. Now can run but not work