def _check_and_set(self, cls_value, sd_value, name):
        """Auxiliary function for checking the values in the checkpoint and
        setting them."""
        if self.override_lr_scheduler:
            print(' > overriding {} value to {}'.format(name, cls_value))
            return cls_value

        if not self.use_checkpoint_lr_scheduler:
            assert cls_value == sd_value, \
                f'AnnealingLR: class input value {cls_value} and checkpoint' \
                f'value {sd_value} for {name} do not match'
        print(' > using checkpoint value {} for {}'.format(sd_value, name))
        return sd_value
Auxiliary function for checking the values in the checkpoint and
        setting them.
training/data_efficiency/gpt_finetuning/learning_rates.py