def set_use_sharded_ddp(self, use_sharded_ddp: bool):
    self.use_sharded_ddp = use_sharded_ddp
    if self.use_sharded_ddp:
        logging.info('Using Sharded DDP')
    return self
def set_use_sharded_ddp(self, use_sharded_ddp: bool):
    self.use_sharded_ddp = use_sharded_ddp
    if self.use_sharded_ddp:
        logging.info('Using Sharded DDP')
    return self