Training on n>1 GPUs (single node), I got https://github.com/Lightning-AI/pytorch-lightning/issues/18803.