diff --git a/python/paddle/trainer_config_helpers/layers.py b/python/paddle/trainer_config_helpers/layers.py index 6b0a8d56cd..b006eb46d9 100755 --- a/python/paddle/trainer_config_helpers/layers.py +++ b/python/paddle/trainer_config_helpers/layers.py @@ -713,9 +713,8 @@ class MixedLayerType(LayerOutput): return self def __exit__(self, exc_type, exc_value, tb): - if exc_type != None: - traceback.print_exception(exc_type, exc_value, tb) - sys.exit(1) + if exc_value is not None: + raise exc_value assert len(self.inputs) != 0 ml = MixedLayer( name=self.name,