hummingbird
/
fastNLP
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

删除训练后的barrier，避免报错后卡住

tags/v1.0.0alpha
x54-729 3 years ago
parent
fcfd8c267e
commit
28b3f10a46
1 changed files with 0 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +0
    -1
      fastNLP/core/controllers/trainer.py

+ 0
- 1
fastNLP/core/controllers/trainer.py View File

@@ -363,7 +363,6 @@ class Trainer(TrainerEventTrigger):
raise e raise e
finally: finally:
self.on_train_end() self.on_train_end()
self.driver.barrier()


def _set_num_eval_batch_per_dl(self, num_eval_batch_per_dl): def _set_num_eval_batch_per_dl(self, num_eval_batch_per_dl):
def _evaluate_fn(trainer: Trainer, evaluate_fn: Callable) -> None: def _evaluate_fn(trainer: Trainer, evaluate_fn: Callable) -> None:


Write Preview
Loading…
Cancel
Save