search

facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
29.79k stars 7.38k forks source link

QAT training support? #2969

Open lucasjinreal opened 3 years ago

lucasjinreal commented 3 years ago

Since some model such MaskRCNN gains a lot speed up and very tiny precesion drop in int8 model, is there any plan to support automatically QAT training support in d2?

zhiqwang commented 3 years ago

Hi, I think that d2go are doing such things.

lucasjinreal commented 3 years ago

@zhiqwang thanks for the information. I saw that most model deploy on mobile devices, does the QAT trained model able to deploy on GPU and accelerate on tensorrt?

zhiqwang commented 3 years ago

For QAT, it seems that d2go only supports faster-rcnn now.

does the QAT trained model able to deploy on GPU and accelerate on tensorrt?

I didn't know either about this.

lucasjinreal commented 3 years ago

@zhiqwang Is that convinient to add QAT on other model trained upon d2 such as SOLOV2?