def step_update(self, num_updates):
        """Called at the end of each step."""
        if (
            self.update_schedule is not None
            and num_updates > 0
            and num_updates % self.update_schedule == 0
        ):
            self.step()
Called at the end of each step.
understand_icl/fairseq/fairseq/quantization_utils.py