Method state_dict

k_diffusion/utils.py:133–135

Returns the state of the class as a :class:`dict`.

(self)

        self.last_epoch = last_epoch

    def state_dict(self):
        """Returns the state of the class as a :class:`dict`."""
        return dict(self.__dict__.items())

    def load_state_dict(self, state_dict):
        """Loads the class's state.