From 6ec566505fe3fdf855f9ee7e93158326a478c5b1 Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Sat, 27 Apr 2024 10:18:48 +0700
Subject: [PATCH] try swin

---
 main.py | 2 ++
 1 file changed, 2 insertions(+)

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: