search

microsoft / GLIP

Grounded Language-Image Pre-training
MIT License
2.07k stars 186 forks source link

Running on CPU #141

Open vatdut8994 opened 9 months ago

vatdut8994 commented 9 months ago

Is there any way to run this code on a CPU? The errors only originate from maskrcnn_benchmark when trying to change the model device to CPU saying that some features are NotImplemented, this is a library made by Facebook quite a while ago, and it says it says that it's deprecated. After doing some research I found that a new version of it has been released named Detectron2. I updated the code accordingly but now the problem lies in the build_detection_model function. Detectron's build_model function seems equivalent, but the GeneralizedVLRCNN apparently cannot be found in META_ARCHITECTURE. Any help concerning this will be very very much appreciated!

Thank you

jy951013 commented 7 months ago

Hi @vatdut8994 , are you able to convert to CPU? Can share with me how you achieve it?

vatdut8994 commented 7 months ago

Hi @vatdut8994 , are you able to convert to CPU? Can share with me how you achieve it?

I quite literally just gave up after 2 weeks of constant trying. Sorry bro