search

wbenbihi / hourglasstensorflow

Tensorflow implementation of Stacked Hourglass Networks for Human Pose Estimation
MIT License
479 stars 177 forks source link

Run program using CPU #25

Closed albertchristianto closed 6 years ago

albertchristianto commented 6 years ago

greetings, I am interested with this project, but I am newbie on tensorflow and I don't GPU support to run on tensorflow-GPU. I can see that there is a code for CPU device, but I have hard time to modify it. is there anyone can help me with it ? At least give me some hint or suggestion to modify the code? Thank you. best regards, Albert

kyrsaexp commented 6 years ago

I guess you can just replace self.gpu in HourglassModel from '/gpu:0' to '/cpu:0'

albertchristianto commented 6 years ago

@usernameis666 really, man. It's work. Haha thanks a lot. I guess this is a silly question.