Method register_state

utils/engine/engine.py:94–95

(self, **kwargs)

        )

    def register_state(self, **kwargs):
        self.state.register(**kwargs)

    def update_iteration(self, epoch, iteration):
        self.state.epoch = epoch