nncf/experimental/torch/nas/bootstrapNAS/training/scheduler.py
def epoch_step(self, next_epoch: Optional[int] = None) -> None:
        """
        Should be called at the beginning of each training epoch to prepare
        the compression method to continue training the model in the `next_epoch`.

        :param next_epoch: The epoch index for which the compression scheduler
            will update the state of the compression method.
        """
        super().epoch_step(next_epoch)
        stage_desc, stage_desc_idx = self.get_current_stage_desc()
        if stage_desc is not None:
            if stage_desc_idx != self.current_stage_idx:
                self._training_ctrl.set_stage(stage_desc)
                self.current_stage_idx = stage_desc_idx

        Should be called at the beginning of each training epoch to prepare
        the compression method to continue training the model in the `next_epoch`.

        :param next_epoch: The epoch index for which the compression scheduler
            will update the state of the compression method.
        
Should be called at the beginning of each training epoch to prepare
the compression method to continue training the model in the `next_epoch`.
Should be called at the beginning of each training epoch to prepare the compression method to continue training the model in the ` next_epoch ` .
def epoch_step(self, next_epoch: Optional[int] = None) -> None:
        super().epoch_step(next_epoch)
        stage_desc, stage_desc_idx = self.get_current_stage_desc()
        if stage_desc is not None:
            if stage_desc_idx != self.current_stage_idx:
                self._training_ctrl.set_stage(stage_desc)
                self.current_stage_idx = stage_desc_idx
def epoch_step ( self , next_epoch : Optional [ int ] = None ) -> None : super ( ) . epoch_step ( next_epoch ) stage_desc , stage_desc_idx = self . get_current_stage_desc ( ) if stage_desc is not None : if stage_desc_idx != self . current_stage_idx : self . _training_ctrl . set_stage ( stage_desc ) self . current_stage_idx = stage_desc_idx
Should be called at the beginning of each training epoch to prepare
the compression method to continue training the model in the `next_epoch`.
Should be called at the beginning of each training epoch to prepare the compression method to continue training the model in the ` next_epoch ` .
"""
        Should be called at the beginning of each training epoch to prepare
        the compression method to continue training the model in the `next_epoch`.

        :param next_epoch: The epoch index for which the compression scheduler
            will update the state of the compression method.
        """
self
next_epoch
Optional[int]
The epoch index for which the compression scheduler
will update the state of the compression method.
The epoch index for which the compression scheduler will update the state of the compression method .