    def set_lr(self, lr):
        """Set the learning rate."""
        for param_group in self.optimizer.param_groups:
            param_group["lr"] = lr
Set the learning rate.
opensora/adaptor/stage_1_and_2.py