19 lines
446 B
Python
19 lines
446 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import time
|
|
|
|
from .hook import HOOKS, Hook
|
|
|
|
|
|
@HOOKS.register_module()
|
|
class IterTimerHook(Hook):
|
|
|
|
def before_epoch(self, runner):
|
|
self.t = time.time()
|
|
|
|
def before_iter(self, runner):
|
|
runner.log_buffer.update({'data_time': time.time() - self.t})
|
|
|
|
def after_iter(self, runner):
|
|
runner.log_buffer.update({'time': time.time() - self.t})
|
|
self.t = time.time()
|