cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

try swin

Browse Source
This commit is contained in:
thanhvc3
2024-04-27 10:18:48 +07:00
parent 30805a0af9
commit 6ec566505f
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -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 utils import get_logger, get_combined_results, set_gpu, prepare_env, set_seed
from models import ComplEx, ConvE, HypER, InteractE, FouriER, TuckER from models import ComplEx, ConvE, HypER, InteractE, FouriER, TuckER
import traceback
class Main(object): class Main(object):
@@ -721,6 +722,7 @@ if __name__ == "__main__":
model.fit() model.fit()
except Exception as e: except Exception as e:
print(e) print(e)
traceback.print_exc()
try: try:
del model del model
except Exception: except Exception: