jiangnana.jnn yingda.chen 3 years ago
parent
commit
f63d7f18f1
1 changed files with 0 additions and 2 deletions
  1. +0
    -2
      modelscope/trainers/trainer.py

+ 0
- 2
modelscope/trainers/trainer.py View File

@@ -828,7 +828,6 @@ class EpochBasedTrainer(BaseTrainer):
self.model.train() self.model.train()
for _ in range(self._epoch, self._max_epochs): for _ in range(self._epoch, self._max_epochs):
self.invoke_hook(TrainerStages.before_train_epoch) self.invoke_hook(TrainerStages.before_train_epoch)
time.sleep(2) # Prevent possible deadlock during epoch transition
for i, data_batch in enumerate(data_loader): for i, data_batch in enumerate(data_loader):
if i < self.inner_iter: if i < self.inner_iter:
# inner_iter may be read out from the checkpoint file, so skip the trained iters in the epoch. # inner_iter may be read out from the checkpoint file, so skip the trained iters in the epoch.
@@ -852,7 +851,6 @@ class EpochBasedTrainer(BaseTrainer):
self._inner_iter = 0 self._inner_iter = 0
self._epoch += 1 self._epoch += 1


time.sleep(1) # wait for some hooks like loggers to finish
self.invoke_hook(TrainerStages.after_run) self.invoke_hook(TrainerStages.after_run)


def evaluation_loop(self, data_loader, metric_classes): def evaluation_loop(self, data_loader, metric_classes):


Loading…
Cancel
Save