From 6d979d8c5a206eed9f48f0ed35098bbc0c01d803 Mon Sep 17 00:00:00 2001
From: revuk100 <renato.vukovic@uni-duesseldorf.de>
Date: Thu, 5 Sep 2024 16:42:29 +0200
Subject: [PATCH] update inference script
---
experiments/TOD_ontology_inference.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/experiments/TOD_ontology_inference.py b/experiments/TOD_ontology_inference.py
index 867d04d..5f0f48d 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"
--
GitLab