修复eta时间计算错误

torchocr/engine/trainer.py

@@ -157,13 +157,12 @@ def train(self):
         total_samples = 0
         train_reader_cost = 0.0
         train_batch_cost = 0.0
-        reader_start = time.time()
         eta_meter = AverageMeter()
 
         for epoch in range(start_epoch, epoch_num + 1):
             if self.train_dataloader.dataset.need_reset:
                 self.train_dataloader = build_dataloader(self.cfg, 'Train', self.logger)
-
+            reader_start = time.time()
             for idx, batch in enumerate(self.train_dataloader):
                 batch = [t.to(self.device) for t in batch]
                 self.optimizer.zero_grad()