running on cpu
Melika-Ayoughi opened this issue · 2 comments
Melika-Ayoughi commented
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
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
Closed due to inactivity.