Skip to content
This repository was archived by the owner on Oct 1, 2020. It is now read-only.
This repository was archived by the owner on Oct 1, 2020. It is now read-only.

Support for nn.ConvTranspose2d #72

Description

@Amitdedhia6

Hi - is there a plan to support nn.ConvTranspose2d for both QNNPACK and FBGEMM? I am using this in my model. After applying QNNPACK quantization techniques (both post training static and QAT), I am getting huge drop in accuracy, as well as not great improvement in speed. I suspect it could be due to ConvTranspose2d.

I am using Unet model for semantic segmentation. In the upsampling side, there are series of layers each having ConvTranspose2d operation. In the quantized model, it has to keep de-quantizing weights/activations to float for ConvTranspose2d layer and then back to quantized form for subsequent convolution/BN/Relu layers.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions