From 994d198eb63136a79b15005e8a9cceec0e521fe4 Mon Sep 17 00:00:00 2001 From: shippingwang Date: Tue, 1 Sep 2020 06:57:24 +0000 Subject: [PATCH] add reader --- tools/program.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tools/program.py b/tools/program.py index 05a14ec97..fe69fd1b2 100644 --- a/tools/program.py +++ b/tools/program.py @@ -314,6 +314,7 @@ def run(dataloader, config, net, optimizer=None, epoch=0, mode='train'): ("lr", AverageMeter( 'lr', 'f', need_avg=False)), ("batch_time", AverageMeter('elapse', '.3f')), + ('reader_time', AverageMeter('reader', '.3f')), ]) else: topk_name = 'top{}'.format(config.topk) @@ -324,10 +325,12 @@ def run(dataloader, config, net, optimizer=None, epoch=0, mode='train'): ("lr", AverageMeter( 'lr', 'f', need_avg=False)), ("batch_time", AverageMeter('elapse', '.3f')), + ('reader_time', AverageMeter('reader', '.3f')), ]) tic = time.time() for idx, batch in enumerate(dataloader()): + metric_list['reader_time'].update(time.time() - tic) batch_size = len(batch[0]) feeds = create_feeds(batch, use_mix) fetchs = create_fetchs(feeds, net, config, mode)