From 0257dc6dde4e13aea632267d5fcdae3a7576db08 Mon Sep 17 00:00:00 2001 From: yh Date: Thu, 6 Jun 2019 00:28:12 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DTrainer=E6=97=A0=E6=B3=95?= =?UTF-8?q?=E6=8D=95=E8=8E=B7Exception=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/core/trainer.py | 2 +- test/core/test_callbacks.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/fastNLP/core/trainer.py b/fastNLP/core/trainer.py index 57a31a69..d7694e00 100644 --- a/fastNLP/core/trainer.py +++ b/fastNLP/core/trainer.py @@ -532,7 +532,7 @@ class Trainer(object): self._train() self.callback_manager.on_train_end() - except Exception as e: + except BaseException as e: self.callback_manager.on_exception(e) if on_exception == 'auto': if not isinstance(e, (CallbackException, KeyboardInterrupt)): diff --git a/test/core/test_callbacks.py b/test/core/test_callbacks.py index e2aa5fa4..71a5565d 100644 --- a/test/core/test_callbacks.py +++ b/test/core/test_callbacks.py @@ -66,8 +66,7 @@ class TestCallback(unittest.TestCase): dev_data=data_set, metrics=AccuracyMetric(pred="predict", target="y"), callbacks=[EarlyStopCallback(5)]) - with self.assertRaises(EarlyStopError): - trainer.train() + trainer.train() def test_lr_scheduler(self): data_set, model = prepare_env()