pspnet training

[kingloo2014]

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]

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

[Da-He]

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.