apaszke:

Secondly, what kind of loss would you like to use exactly? If it’s segmentation then, nn.NLLLoss2d should do it.


I have some co-ordinates for bounding boxes in the train dataset and I want the net to train on those.