try swin

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: