Checking Duration in torch.cuda

import torchclass Event_record(): def __init__(self): self.start = torch.cuda.Event(enable_timing=True) self.end = torch.cuda.Event(enable_timing=True) def record_start(self): self.start.record() def record_end(self): self.end.record() def __enter__(self): self.record_start() def __exit__(self, exc_type, exc_value, traceback): self.record_..

2024.02.07