Closed xdcui-nlp closed 2 years ago
Hello, I'm surprised. What model is model1? The results are the same as cronkgqa
QA_model_EmbedKGQA_complex
is the actual EmbedKGQA model. What do you mean by model1?
QA_model_EmbedKGQA_complex
is the actual EmbedKGQA model. What do you mean by model1?
Because I trained embedkgqa and model1 and got the same results, I don't know which model model1 represents
Can you elaborate how you trained those two models? ie the commands used?
Can you elaborate how you trained those two models? ie the commands used?
I used the commands of running code in readme
Are these the exact commands you used?
CUDA_VISIBLE_DEVICES=1 python -W ignore ./train_qa_model.py --frozen 1 --eval_k 1 --max_epochs 200 \
--lr 0.00002 --batch_size 250 --mode train --tkbc_model_file tcomplex_17dec.ckpt \
--dataset wikidata_big --valid_freq 3 --model model1 --valid_batch_size 50 \
--save_to temp --lm_frozen 1 --eval_split valid
and
CUDA_VISIBLE_DEVICES=1 python -W ignore ./train_qa_model.py --frozen 1 --eval_k 1 --max_epochs 200 \
--lr 0.00002 --batch_size 250 --mode train --tkbc_model_file tcomplex_17dec.ckpt \
--dataset wikidata_big --valid_freq 3 --model embedkgqa --valid_batch_size 50 \
--save_to temp --lm_frozen 1 --eval_split valid
Also can you show what results you got with model1? In my experiments with the base model (model1) I do not get as good results as CronKGQA. It would be helpful if you could share the exact command you ran
Also can you show what results you got with model1? In my experiments with the base model (model1) I do not get as good results as CronKGQA. It would be helpful if you could share the exact command you ran
yes,I used these commands, but I did't save the results of model1, I can train it again, then I can provide the results to you.Thank you very much for answering my question
hi,I have another question,I trained embedkgqa_complex,but when I to eval it , the following error occurred:
Traceback (most recent call last):
File "./train_qa_model.py", line 551, in
This is because embedkgqa_complex needs non-temporal complex embeddings. You can train those separately, but we will upload a trained checkpoint soon. It would be great if you could create a separate issue for this
This is because embedkgqa_complex needs non-temporal complex embeddings. You can train those separately, but we will upload a trained checkpoint soon. It would be great if you could create a separate issue for this
ok, thank you
hi,embedkgqa model = cronkgqa,but which model is the real embedkgqa?