Method set_init_step

cdslib/core/optim/tf_optim.py:111–113

Set the initial step.

(self, init_step: int)

        self.use_dynamic_rate = use_dynamic_rate

    def set_init_step(self, init_step: int):
        """Set the initial step."""
        self._step = init_step

    def zero_grad(self, set_to_none: bool = False):
        """Zero the gradient buffer."""