search

erikwijmans / Pointnet2_PyTorch

PyTorch implementation of Pointnet2/Pointnet++
The Unlicense
1.5k stars 340 forks source link

Loss function #104

Closed MCLYang closed 4 years ago

MCLYang commented 4 years ago

Hi @erikwijmans, I am so new to pytorch-lighting. I did not find the loss function from the code of trainer. What is the loss function for the semantic segmentation? From other implementation for pointnet++, I found its just like F.nll_loss() but I still want to confirm if your version is using F.nll_loss() or you add the regularizer?

erikwijmans commented 4 years ago

The training step for all models is here: https://github.com/erikwijmans/Pointnet2_PyTorch/blob/master/pointnet2/models/pointnet2_ssg_cls.py#L125

I am using F.cross_entropy (which is softmax + nll_loss).

MCLYang commented 4 years ago

thank you