Skip to content

Commit 096f10c

Browse files
martinlsmMartin Lindströmzingo
authored
Arm backend: Make aot_arm_compiler.py functions importable (#18039)
Calibration was tied to the ImageNet evaluator, so when the evaluation feature was removed from aot_arm_compiler.py, custom calibration of a quantized model was no longer possible. This patch reimplements quantization calibration into aot_arm_compiler.py, but it is instead controlled by a flag called --calibration_data. This flag can target a file that is either a common image format, a serialized file (.pt), or a directory which will make the program search for applicable files within the directory. cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell --------- Signed-off-by: Martin Lindström <Martin.Lindstroem@arm.com> Co-authored-by: Martin Lindström <Martin.Lindstroem@arm.com> Co-authored-by: Zingo Andersen <zingo.andersen@arm.com>
1 parent 2b868cb commit 096f10c

1 file changed

Lines changed: 221 additions & 51 deletions

File tree

0 commit comments

Comments
 (0)