[PyTorch] ValueError: pos_label=1 is not a valid label: array([ 0., 255.])