fix labeling loss layer