def recalculate(self, loader_batch_len: int, current_batch_step: int) -> None:
        """Recalculates total num_steps for ``batch`` mode.

        Args:
            loader_batch_len: total count of batches in an epoch
            current_batch_step: current step
        """
        warmup_steps = self.warmup_steps * loader_batch_len
        lr_annealing_steps = self.lr_annealing_steps * loader_batch_len
        decay_steps = self.decay_steps * loader_batch_len

        self._calculate_lr_momentum(warmup_steps, lr_annealing_steps, decay_steps)
        self.last_epoch = current_batch_step * loader_batch_len
Recalculates total num_steps for ``batch`` mode.

        Args:
            loader_batch_len: total count of batches in an epoch
            current_batch_step: current step
        
catalyst/contrib/schedulers/onecycle.py