fix ips print error

whutbd · Aug 29, 2023 · 8b780d8 · 8b780d8
1 parent 6bdef6a
commit 8b780d8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/configs/rec/PP-OCRv3/ch_PP-OCRv3_rec.yml b/configs/rec/PP-OCRv3/ch_PP-OCRv3_rec.yml
@@ -97,7 +97,7 @@ Train:
     shuffle: true
     batch_size_per_card: 128
     drop_last: true
-    num_workers: 0
+    num_workers: 4
 Eval:
   dataset:
     name: SimpleDataSet

diff --git a/torchocr/engine/trainer.py b/torchocr/engine/trainer.py
@@ -189,14 +189,14 @@ def train(self):
                     metric = self.eval_class.get_metric()
                     train_stats.update(metric)
 
-                self.lr_scheduler.step()
-
                 train_batch_time = time.time() - reader_start
                 train_batch_cost += train_batch_time
                 eta_meter.update(train_batch_time)
                 global_step += 1
                 total_samples += len(batch[0])
 
+                self.lr_scheduler.step()
+
                 # logger
                 stats = {
                     k: float(v) if v.shape == [] else v.detach().cpu().numpy().mean()
@@ -223,11 +223,10 @@ def train(self):
                             f'ips: {total_samples / train_batch_cost:.5f} samples/s, '
                             f'eta: {eta_sec_format}')
                     self.logger.info(strs)
-
                     total_samples = 0
                     train_reader_cost = 0.0
                     train_batch_cost = 0.0
-
+                reader_start = time.time()
             if self.local_rank == 0:
                 save_ckpt(self.model, self.cfg, self.optimizer, self.lr_scheduler, epoch, global_step, best_metric,
                           is_best=False)
@@ -254,7 +253,6 @@ def train(self):
                                                best_metric[self.eval_class.main_indicator], global_step)
                 best_str = f"best metric, {', '.join(['{}: {}'.format(k, v) for k, v in best_metric.items()])}"
                 self.logger.info(best_str)
-            reader_start = time.time()
         best_str = f"best metric, {', '.join(['{}: {}'.format(k, v) for k, v in best_metric.items()])}"
         self.logger.info(best_str)
         if self.writer is not None: