How to export FP32 model?
JingruiYu opened this issue · 2 comments
JingruiYu commented
EVALUATION.md shows the latency of FP32. With export.py and optimize.py only the FP16 model is available. So how to export FP32 model? And could you give the accuracy of different models? thanks!
fabio-sim commented
Hi @JingruiYu, thank you for your interest in LightGlue-ONNX.
If you run optimize.py
as follows:
python optimize.py --input weights/superpoint_lightglue.onnx
It should produce two outputs:
weights/superpoint_lightglue_fused.onnx
: This is the FP32 oneweights/superpoint_lightglue_fused_fp16.onnx
: This is the FP16 one
At the moment, I don't have any plans to calculate the accuracies (busy with other stuff). But you're very welcome to contribute these results!
JingruiYu commented
Thanks!