Skip to content

Commit 4d700c4

Browse files
kimishpatelfacebook-github-bot
authored andcommitted
enable vit quantization
Summary: After 0919 nightly of pytorch vit linear layers should be quantized Differential Revision: D49438891
1 parent 052f624 commit 4d700c4

File tree

1 file changed

+1
-0
lines changed
  • examples/recipes/xnnpack_optimization

1 file changed

+1
-0
lines changed

examples/recipes/xnnpack_optimization/models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,4 +24,5 @@ class OptimizationOptions(object):
2424
), # TODO[T163161310]: takes a long time to export to exec prog and save inception_v4 quantized model
2525
"w2l": OptimizationOptions(False, True),
2626
"dl3": OptimizationOptions(True, False),
27+
"vit": OptimizationOptions(True, False),
2728
}

0 commit comments

Comments
 (0)