forked from open-mmlab/mmrazor
-
Notifications
You must be signed in to change notification settings - Fork 2
/
ptq_openvino_yolox_s_8xb8-300e_coco_calib32xb32.py
57 lines (50 loc) · 1.85 KB
/
ptq_openvino_yolox_s_8xb8-300e_coco_calib32xb32.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
_base_ = [
'mmdet::yolox/yolox_s_8xb8-300e_coco.py',
'../../deploy_cfgs/mmdet/detection_openvino_dynamic-800x1344.py'
]
_base_.val_dataloader.batch_size = 32
test_cfg = dict(
type='mmrazor.PTQLoop',
calibrate_dataloader=_base_.val_dataloader,
calibrate_steps=32,
)
float_checkpoint = 'https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_s_8x8_300e_coco/yolox_s_8x8_300e_coco_20211121_095711-4592a793.pth' # noqa: E501
global_qconfig = dict(
w_observer=dict(type='mmrazor.PerChannelMinMaxObserver'),
a_observer=dict(type='mmrazor.MovingAverageMinMaxObserver'),
w_fake_quant=dict(type='mmrazor.FakeQuantize'),
a_fake_quant=dict(type='mmrazor.FakeQuantize'),
w_qscheme=dict(
qdtype='qint8', bit=8, is_symmetry=True, is_symmetric_range=True),
a_qscheme=dict(qdtype='quint8', bit=8, is_symmetry=True),
)
model = dict(
_delete_=True,
type='mmrazor.MMArchitectureQuant',
data_preprocessor=dict(
type='mmdet.DetDataPreprocessor',
pad_size_divisor=32,
batch_augments=[
dict(
type='mmdet.BatchSyncRandomResize',
random_size_range=(480, 800),
size_divisor=32,
interval=10)
]),
architecture=_base_.model,
deploy_cfg=_base_.deploy_cfg,
float_checkpoint=float_checkpoint,
quantizer=dict(
type='mmrazor.OpenVINOQuantizer',
global_qconfig=global_qconfig,
tracer=dict(
type='mmrazor.CustomTracer',
skipped_methods=[
'mmdet.models.dense_heads.yolox_head.YOLOXHead.predict_by_feat', # noqa: E501
'mmdet.models.dense_heads.yolox_head.YOLOXHead.loss_by_feat',
])))
model_wrapper_cfg = dict(
type='mmrazor.MMArchitectureQuantDDP',
broadcast_buffers=False,
find_unused_parameters=True)
custom_hooks = []