Can this project work with CPU?

INK-USC / RE-Net

Recurrent Event Network: Autoregressive Structure Inference over Temporal Knowledge Graphs (EMNLP 2020)

http://inklab.usc.edu/renet/

435 stars 95 forks source link

Can this project work with CPU? #44

Closed mscsedu closed 3 years ago

mscsedu commented 3 years ago

As the title, can this project work with CPU as I don't have GPU. Moreover, what will be the estimated performance?

mscsedu commented 3 years ago

Secondly, For TATransE, there are no files created during the data preprocessing of data named train2id.txt, valid2id.txt, test2id.txt. These files are called in TATransE in line 142. Could you please guide me about these files? Or they simply the train, test, and valid file?

woojeongjin commented 3 years ago

Unfortunately, the code does not work with CPU.

For TATransE, you should run the following commands

cd data
python3 preprocess_TA_step1.py ICEWS18
python3 preprocess_TA_step2.py ICEWS18