RandLA-Net-pytorch icon indicating copy to clipboard operation
RandLA-Net-pytorch copied to clipboard

why needs torch.distributions.utils.probs_to_logits?

Open tangyipeng100 opened this issue 3 years ago • 0 comments

I have noticed that there comes torch.distributions.utils.probs_to_logits after forward propagation:

train.py 115~117:

scores = model(points)#[b, 40960, 6]->[b, 14, 40960]

logp = torch.distributions.utils.probs_to_logits(scores, is_binary=False)

why needs torch.distributions.utils.probs_to_logits? Can I remove it and use scores directly to calculate the loss?

tangyipeng100 avatar Aug 10 '21 07:08 tangyipeng100