i really don’t get what you want to do.
what is uncorrelated_c_pairs?
also , i think the below  part will riase an error OR it’s really hard to read



 ailneni_rao_19210068:

return torch.mean(((weights[:,0]*(1-targets))+(weights[:,1]*targets))torch.nn.BCEWithLogitsLoss()(outputs, targets),axis=1)+0.05corr_loss/len(uncorrelated_c_pairs)


what is mathematical expression for the loss?