NeuralTripleTranslation icon indicating copy to clipboard operation
NeuralTripleTranslation copied to clipboard

How will I train the model and get the pickle files

Open archana3194 opened this issue 5 years ago • 6 comments

IN WHICH FOLDER I NEED TO KEEP THE WIKI-DBPEDIA_TRAIN.PY @YueLiu

archana3194 avatar Mar 10 '19 04:03 archana3194

Could you please answer this?

lalppaul avatar Jul 08 '19 21:07 lalppaul

Hi @archana3194 @lalppaul , there were two tf generated files you got after training the model. because they were too large so I could not upload them. you should be able to get them once you ran the end-end training. I suggest you try to use seq2seq/s2s_preprocess.py to generate the .pkl file. there s a local path needed for pretrained word2vec. Then you may try baselines/main.py and see if your input dimension is valid. (you might need to swap the loss.data[0] to loss.item() in main.py due to the recent update in pytorch)

YueLiu avatar Jul 08 '19 21:07 YueLiu

and of course, you need to prepare your training corpus following this pattern

YueLiu avatar Jul 08 '19 21:07 YueLiu

Thanks for the response. I ran the s2s_preprocess.py file and generated the 20180405.pkl file. After that i tried running the main.py from baselines and found out that i was missing train, test and valid pkl files. Am i missing any steps in between. Also, could you please provide the training corpus? Thanks in advance!!

lalppaul avatar Jul 09 '19 17:07 lalppaul

@YueLiu I have the same problem - could you please update the instruction how to run your code?

petrLorenc avatar Nov 05 '19 17:11 petrLorenc

Has anyone gotten this code working that could give the steps required to get it training?

mitchelldehaven avatar Feb 12 '20 21:02 mitchelldehaven