Error when using DDP and deep_supervision=False

Only last conv layer in seg_layers will be used when deep_supervision=False. However DDP expects there are no unused parameters in the model. PyTorch version 2.5

RuntimeError: Expected to have finished reduction in the prior iteration before starting a new one. Thi s error indicates that your module has parameters that were not used in producing loss.