rail_marking
rail_marking copied to clipboard
OHEMCELoss called twice
Is there any reason OHEMCELoss called twice here? https://github.com/xmba15/rail_marking/blob/6372aa586ae77a7f4a0fe0e1b3e57ce1213a911b/scripts/segmentation/train.py#L121-L122 Thank you for your precious time on my question.