Actually, I´m not quite sure of what lightning is using. I tried to set the flag  accelerator=“ddp” on the Trainer but the problem persists.
Could I force “ddp_model = DDP(model, device_ids=[rank])” over my model or may that results in conflict in lightning?