Changing Model Architecture after DDP

found my solution here Access to attributes of model wrapped in DDP