Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-02 12:09:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933]remove sleep in train_loop

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9910419
This commit is contained in:
jiangnana.jnn
2022-10-13 10:39:56 +08:00
committed by yingda.chen
parent 144ffee2cf
commit f63d7f18f1
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/trainers/trainer.py
View File

@@ -828,7 +828,6 @@ class EpochBasedTrainer(BaseTrainer):
self.model.train()
for _ in range(self._epoch, self._max_epochs):
self.invoke_hook(TrainerStages.before_train_epoch)
time.sleep(2) # Prevent possible deadlock during epoch transition
for i, data_batch in enumerate(data_loader):
if i < self.inner_iter:
# 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._epoch += 1
time.sleep(1) # wait for some hooks like loggers to finish
self.invoke_hook(TrainerStages.after_run)
def evaluation_loop(self, data_loader, metric_classes):