I’m a bit stumped here. I have a semantic segmentation model that trains fine on a single gpu, but when I try dataparallel with two my loss increases until I get NaN. Any common pitfalls that I should watch out for with dataparallel and semantic segmentation?