Skip to content
Snippets Groups Projects
Unverified Commit 6adc01ed authored by Haian Huang(深度眸)'s avatar Haian Huang(深度眸) Committed by GitHub
Browse files

Add cpu compatibility info in documentation (#4405)

* Add cpu compatibility info

* Add table about not support CPU only model info

* Update docs about not support CPU only model info
parent b3c67d5e
No related branches found
No related tags found
No related merge requests found
...@@ -118,12 +118,28 @@ In CPU mode you can run the demo/webcam_demo.py for example. ...@@ -118,12 +118,28 @@ In CPU mode you can run the demo/webcam_demo.py for example.
However some functionality is gone in this mode: However some functionality is gone in this mode:
- Deformable Convolution - Deformable Convolution
- Modulated Deformable Convolution
- ROI pooling
- Deformable ROI pooling - Deformable ROI pooling
- CARAFE: Content-Aware ReAssembly of FEatures - CARAFE: Content-Aware ReAssembly of FEatures
- SyncBatchNorm
- CrissCrossAttention: Criss-Cross Attention
- MaskedConv2d
- Temporal Interlace Shift
- nms_cuda - nms_cuda
- sigmoid_focal_loss_cuda - sigmoid_focal_loss_cuda
- bbox_overlaps
So if you try to run inference with a model containing deformable convolution you will get an error. So if you try to run inference with a model containing above ops you will get an error. The following table lists the related methods that cannot inference on CPU due to dependency on these operators
| Operator | Model |
| :-----------------------------------------------------: | :----------------------------------------------------------: |
| Deformable Convolution/Modulated Deformable Convolution | DCN、Guided Anchoring、RepPoints、CentripetalNet、VFNet、CascadeRPN、NAS-FCOS、DetectoRS |
| MaskedConv2d | Guided Anchoring |
| CARAFE | CARAFE |
| SyncBatchNorm | ResNeSt |
**Notice**: MMDetection does not support training with CPU for now.
### Another option: Docker Image ### Another option: Docker Image
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment