7 ms·
That would be awesome. But - is a tutorial enough? From what I gather from the readme the custom ops have cuda implementations and no CPU ones, which you'd like
by haeffin 9y ago
That would be awesome. But - is a tutorial enough? From what I gather from the readme the custom ops have cuda implementations and no CPU ones, which you'd like to have as a fallback on mobile (even if metal/opengl implementations exist), or am I wrong?
- ajtulloch 9y agoThey exist and are (nearly all) open-sourced for CPU & Metal (I wrote some of them).
- haeffin 9y agoNice - the "Detectron operators currently do not have CPU implementation; a GPU system is required." on https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md https://github.com/facebookresearch/Detectron/blob/master/IN... is outdated then? Or is it the "nearly all" that's behind this statement?
- stzpz 9y agoWe have open sourced several operators for detectron so that the model could run on GPU/CPU/mobile (https://github.com/facebookresearch/Detectron/commit/757d77c239fbef9d6a341cdfcea2f43219175709 https://github.com/facebookresearch/Detectron/commit/757d77c...). We also open sourced a conversion tool (https://github.com/facebookresearch/Detectron/commit/757d77c239fbef9d6a341cdfcea2f43219175709 https://github.com/facebookresearch/Detectron/commit/757d77c...) to help converting the model.