-
Notifications
You must be signed in to change notification settings - Fork 7.1k
Open
Labels
module: models.quantizationIssues related to the quantizable/quantized modelsIssues related to the quantizable/quantized models
Description
🚀 The feature
Add quantization support for BackboneWithFPN
.
Motivation, pitch
Currently, it is possible to use from torchvision.models.detection.backbone_utils.resnet_fpn_backbone/BackboneWithFPN
in order to produce a backbone fpn network, given some sort of network that can produce features.
While the original resnet/other base networks you want to use (ex. efficientnet), have quantized variants supported (with pretrained weights too), it would be nice to support QAT in the context of FPN networks, to speed up the detectors even more.
Alternatives
No response
Additional context
No response
zhiqwang
Metadata
Metadata
Assignees
Labels
module: models.quantizationIssues related to the quantizable/quantized modelsIssues related to the quantizable/quantized models