diff --git a/main.py b/main.py index 31892b0..f6dea4c 100644 --- a/main.py +++ b/main.py @@ -20,6 +20,7 @@ from data_loader import TrainDataset, TestDataset from utils import get_logger, get_combined_results, set_gpu, prepare_env, set_seed from models import ComplEx, ConvE, HypER, InteractE, FouriER, TuckER +import traceback class Main(object): @@ -721,6 +722,7 @@ if __name__ == "__main__": model.fit() except Exception as e: print(e) + traceback.print_exc() try: del model except Exception: