From f144ff0cc372017a4979a2c609c3f110eb941a28 Mon Sep 17 00:00:00 2001 From: Alex Amati Date: Mon, 10 Nov 2025 19:09:53 +0100 Subject: [PATCH] Fixed incorrect batch times during training --- lib/core/function.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/lib/core/function.py b/lib/core/function.py index 554b1fb388..dad054e7f9 100644 --- a/lib/core/function.py +++ b/lib/core/function.py @@ -47,10 +47,8 @@ def train(cfg, train_loader, model, criterion, optimizer, scaler, epoch, num_bat # switch to train mode model.train() - start = time.time() for i, (input, target, paths, shapes) in enumerate(train_loader): - intermediate = time.time() - #print('tims:{}'.format(intermediate-start)) + start = time.time() num_iter = i + num_batch * (epoch - 1) if num_iter < num_warmup: @@ -507,4 +505,4 @@ def update(self, val, n=1): self.val = val self.sum += val * n self.count += n - self.avg = self.sum / self.count if self.count != 0 else 0 \ No newline at end of file + self.avg = self.sum / self.count if self.count != 0 else 0