search

linjieli222 / HERO

Research code for EMNLP 2020 paper "HERO: Hierarchical Encoder for Video+Language Omni-representation Pre-training"
https://arxiv.org/abs/2005.00200
MIT License
230 stars 34 forks source link

running on cpu #3

Closed Melika-Ayoughi closed 3 years ago

Melika-Ayoughi commented 3 years ago

Is it possible to run the pretrained model on cpu? I see the requirement for this project is nvidia (for gpus) but I have a MacBook and it does not have a gpu.

linjieli222 commented 3 years ago

Sorry for the late reply and thanks for your interests in this project.

Our code is built on GPU, without the support on CPU. However, if you insists on running it on CPU, you can modify the code to disable horovod, apex and set the device to CPU. I believe there are some works to be done to make it happen. But let me know if you encounter any unexpected issues.

Thanks, Linjie

linjieli222 commented 3 years ago

Closed due to inactivity.