Skip to content

Commit

Permalink
Enable quantization for wav2letter (#446)
Browse files Browse the repository at this point in the history
Summary:

X-link: pytorch/pytorch#109830

Also added annotation support for conv1d_relu and conv1d in XNNPACKQuantizer, the quantized results still
matches fx quant path (didn't quantize conv1d) so tests are not disabled

Reviewed By: kimishpatel

Differential Revision: D49479546
  • Loading branch information
jerryzh168 authored and facebook-github-bot committed Sep 28, 2023
1 parent 732d92b commit 0ac9d5d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/recipes/xnnpack_optimization/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class OptimizationOptions(object):
"resnet18": OptimizationOptions(True, True),
"resnet50": OptimizationOptions(True, True),
"vit": OptimizationOptions(False, True),
"w2l": OptimizationOptions(False, True),
"w2l": OptimizationOptions(True, True),
"edsr": OptimizationOptions(True, False),
"mobilebert": OptimizationOptions(True, False),
}

0 comments on commit 0ac9d5d

Please sign in to comment.