Shen

The same problem. Fixed it by changing every hp_mask = (label.unsqueeze(1) == label.unsqueeze(2)).byte().view(-1) hn_mask = (label.unsqueeze(1) != label.unsqueeze(2)).byte().view(-1) to hp_mask = (label.unsqueeze(1) == label.unsqueeze(2)).bool().view(-1) hn_mask = (label.unsqueeze(1) != label.unsqueeze(2)).bool().view(-1) in...