修复Trainer无法捕获Exception的bug

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)):

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()