PyCDA
PyCDA copied to clipboard
aux_loss
Hi,
You use two losses for your training, yet all your decoders return the main output only, the aux one being None.
Which one should I add ?
Maybe I misunderstood, but doesn't _ discard the variable ?
Hi @Xcrid , sorry for the late reply. The aux loss is from the PSPNet. the implementation of PSPNet is from https://github.com/CSAILVision/semantic-segmentation-pytorch. Maybe It is confused for the variable name. I will modify it once I have time.