Method register

utils/engine/engine.py:30–33

(self, **kwargs)

        self.optimizer = None

    def register(self, **kwargs):
        for k, v in kwargs.items():
            assert k in ["epoch", "iteration", "dataloader", "model", "optimizer"]
            setattr(self, k, v)


class Engine(object):