def METHOD_NAME(self, lr_scheduler: LRScheduler, checkpoint: str):
    """
    Save model to checkpoint but only on master process.
    """
    if self.coordinator.is_master():
        super().METHOD_NAME(lr_scheduler, checkpoint)