Method __exit__

utils/engine/engine.py:185–189

(self, type, value, tb)

        return self

    def __exit__(self, type, value, tb):
        torch.cuda.empty_cache()
        if type is not None:
            logger.warning("A exception occurred during Engine initialization, give up running process")
            return False