diff --git a/experiments/TOD_ontology_inference.py b/experiments/TOD_ontology_inference.py index 867d04d9bef372db60b872f8f2da98a81d2c171f..5f0f48de458f22c2588e4251b42fa6d8b11bc29e 100644 --- a/experiments/TOD_ontology_inference.py +++ b/experiments/TOD_ontology_inference.py @@ -178,7 +178,8 @@ def main(): except Exception as e: logger.info(f"Checkpoint saved at {checkpoint_filename} after {counter} dialogues") logger.error(f"Error at dialogue {dial_id} in split {split}") - continue + logger.error(f"Error message: {e}") + e.with_traceback() if config.predict_for_cot_decoding: output_string += "Step " + str(i) + " response:\n"