meetps/pytorch-semseg

pspnet training

kingloo2014 opened this issue · 2 comments

kingloo2014 commented

hello, when I trained the pspnet, I found the ouput of this network has 2 tensors. So, the loss function cannot accept this output and lead to crash. Could you tell me which loss function should be used in this network ?

yyyjia commented

When training, there are two loss functions, only one in the test phase.

Da-He commented

hello, when I trained the pspnet, I found the ouput of this network has 2 tensors. So, the loss function cannot accept this output and lead to crash. Could you tell me which loss function should be used in this network ?

When using PSPNet, it seems that you should use "multi_scale_cross_entropy" instead of "cross_entropy" in your config file.

  • 👍1