From f17343e19b8c1070b98a82dc2a0e7bb0622f7572 Mon Sep 17 00:00:00 2001 From: ROGERDJQ Date: Wed, 16 Dec 2020 18:04:22 +0800 Subject: [PATCH] =?UTF-8?q?[bugfix]=E4=BF=AE=E5=A4=8Dfitlogcallback?= =?UTF-8?q?=E5=9C=A8disttrainner=E4=B8=AD=E6=97=A0=E6=B3=95=E6=B7=BB?= =?UTF-8?q?=E5=8A=A0dev=5Fdata=20=E7=9A=84=E9=97=AE=E9=A2=98=20(#348)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit fix the distTrainer dev_data --- fastNLP/core/dist_trainer.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/fastNLP/core/dist_trainer.py b/fastNLP/core/dist_trainer.py index 35970581..a26673b2 100644 --- a/fastNLP/core/dist_trainer.py +++ b/fastNLP/core/dist_trainer.py @@ -177,8 +177,13 @@ class DistTrainer(): self.batch_size = self.world_size * self.batch_size_per_gpu self.n_steps = self._get_n_steps() + self.dev_data = dev_data + self.metrics = metrics + self.test_use_tqdm = True + self.kwargs = kwargs self.test_use_tqdm = kwargs.get('test_use_tqdm', self.use_tqdm) dev_batch_size = kwargs.get('dev_batch_size', batch_size_per_gpu) + # for evaluation, only run eval on master proc if dev_data and metrics: cb = _TesterCallback(