Commit 00e8a4f0 authored by Peizhao Zhang's avatar Peizhao Zhang Committed by Facebook GitHub Bot
Browse files

Skipped broken unit tests for exporter.

Summary:
Skipped broken unit tests for exporter.
* The qat exporter test is not working. See T85959661

Reviewed By: zhanghang1989

Differential Revision: D26785458

fbshipit-source-id: 813316e110698eeeb0b3afdf93d97b30f0b86508
parent 66b7c7c8
...@@ -57,11 +57,20 @@ def _apply_eager_mode_quant(cfg, model): ...@@ -57,11 +57,20 @@ def _apply_eager_mode_quant(cfg, model):
n_outputs=len(cfg.MODEL.RPN.IN_FEATURES) * 2, n_outputs=len(cfg.MODEL.RPN.IN_FEATURES) * 2,
) )
# Wrap the roi_heads, box_pooler is not quantized # Wrap the roi_heads, box_pooler is not quantized
if hasattr(model.roi_heads, "box_head"):
model.roi_heads.box_head = wrap_quant_subclass( model.roi_heads.box_head = wrap_quant_subclass(
model.roi_heads.box_head, model.roi_heads.box_head,
n_inputs=1, n_inputs=1,
n_outputs=1, n_outputs=1,
) )
# for faster_rcnn_R_50_C4
if hasattr(model.roi_heads, "res5"):
model.roi_heads.res5 = wrap_quant_subclass(
model.roi_heads.res5,
n_inputs=1,
n_outputs=1,
)
model.roi_heads.box_predictor = wrap_quant_subclass( model.roi_heads.box_predictor = wrap_quant_subclass(
model.roi_heads.box_predictor, n_inputs=1, n_outputs=2 model.roi_heads.box_predictor, n_inputs=1, n_outputs=2
) )
...@@ -116,4 +125,3 @@ def d2_meta_arch_prepare_for_quant(self, cfg): ...@@ -116,4 +125,3 @@ def d2_meta_arch_prepare_for_quant(self, cfg):
logger.info("Setup the model with qconfig:\n{}".format(model.qconfig)) logger.info("Setup the model with qconfig:\n{}".format(model.qconfig))
return model return model
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment