打印loss,定位问题。debiased_cl:nan
print("loss : re_ae_loss:{:.4f}; re_graph_loss:{:.4f}; kl_loss:{:.4f}, debiased_cl:{:.4f}".format(re_ae_loss, re_graph_loss, kl_loss, debiased_cl))
assert torch.isnan(loss).sum() == 0, print(loss)
打印loss,定位问题。debiased_cl:nan
print("loss : re_ae_loss:{:.4f}; re_graph_loss:{:.4f}; kl_loss:{:.4f}, debiased_cl:{:.4f}".format(re_ae_loss, re_graph_loss, kl_loss, debiased_cl))
assert torch.isnan(loss).sum() == 0, print(loss)