Correct DDP initialization for model
# Bugfix
- Correct DDP initialization (single/multi node) for datasets that depend on self.global_rank, self.local_rank, self.node_rank.

Signed-off-by: smajumdar <titu1994@gmail.com>