@property
def model(self):
    'Returns model used in training (can be wrapped with DDP)'
    return (self.distributed_model if is_distributed_training_run() else self.base_model)
Returns model used in training (can be wrapped with DDP)
@property
def model(self):
    
    return (self.distributed_model if is_distributed_training_run() else self.base_model)