Ultra-Fast-Lane-Detection-v2 icon indicating copy to clipboard operation
Ultra-Fast-Lane-Detection-v2 copied to clipboard

关于模型大小的问题,删除optimizer之后为何模型还有上百兆的空间,转化为onnx后仍然很大,请问这是正常的吗?

Open VanlenFrank opened this issue 1 year ago • 8 comments

VanlenFrank avatar Aug 18 '23 15:08 VanlenFrank

image image 这两张图分别是项目中提供的预训练模型大小,和加载之后看到的net中保存的内容(只有model,没有optimizer),依然有上百兆。如果集成到端侧,模型太大每次OTA升级需要很久,求大佬指点 @cfzd

VanlenFrank avatar Aug 18 '23 15:08 VanlenFrank

image 这个结构

umie0128 avatar Aug 23 '23 06:08 umie0128

@umie0128 什么意思,FC占用了大量的空间?如果是这样,也没办去掉吧,这是模型的一部分

VanlenFrank avatar Aug 24 '23 02:08 VanlenFrank

自问自答:网络比较打的原因两个,一是FC占用了很大的空间,如果将第一层的FC中2048降低了512,网络结构回降低到三分之一,当然性能也随之下降(但是在tusimple上下降不大)。二是在保存网络参数的时候同事保存了state_dict和optimizer,去掉optimizer会下降一半。

VanlenFrank avatar Aug 30 '23 05:08 VanlenFrank

可以优化后面的两个全连接,先split分开fc

licc0431 avatar Oct 18 '23 01:10 licc0431

您发的邮件已收到,谢谢——武龙

VanlenFrank avatar Oct 18 '23 01:10 VanlenFrank

你好 可以加一个联系方式嘛 关于这个模型有些问题想请教一下

Yysheepsheep avatar Apr 25 '24 12:04 Yysheepsheep

您发的邮件已收到,谢谢——武龙

VanlenFrank avatar Apr 25 '24 12:04 VanlenFrank