search

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

How to train without GPU #13

Closed martozzz closed 5 years ago

martozzz commented 5 years ago

As we only have multi-core CPUs available currently, can we train without GPUs?

woojeongjin commented 5 years ago

In the current code, no. But you can run without GPU by changing all GPU tensors to CPU tensors in the code.

Thanks!

shanzhenren commented 5 years ago

Hi @MartinZZZ, @woojeongjin will push an update on the script with CPU training option, even though we highly recommend to do GPU training for efficiency consideration.