deepinsight / deepinsight/insightface

torch2onnx file size

Open
#2,292 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
29.7k
Forks
6.1k
PR merge metrics
No merged PRs in 30d

Description

I recently trained a wf42m-vitB-pfc03 model using the wf42m_pfc03_40epoch_8gpu_vit_b config. It came out as 444MB. After converting to onnx using the provided [torch2onnx](https://github.com/deepinsight/insightface/blob/master/recognition/arcface_torch/torch2onnx.py) script, it came out slightly larger in size. When using the `simplify` argument it became slightly bigger than without.

Is this suppose to happen?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.