Method activate

classification/ema_deepspeed.py:94–100

(self, model)

    @contextmanager
    def activate(self, model):
        try:
            self.store(model)
            self.copy_to(model)
            yield
        finally:
            self.restore(model)