Ultra-Fast-Lane-Detection-v2
Ultra-Fast-Lane-Detection-v2 copied to clipboard
关于模型大小的问题,删除optimizer之后为何模型还有上百兆的空间,转化为onnx后仍然很大,请问这是正常的吗?
这两张图分别是项目中提供的预训练模型大小,和加载之后看到的net中保存的内容(只有model,没有optimizer),依然有上百兆。如果集成到端侧,模型太大每次OTA升级需要很久,求大佬指点 @cfzd
这个结构
@umie0128 什么意思,FC占用了大量的空间?如果是这样,也没办去掉吧,这是模型的一部分
自问自答:网络比较打的原因两个,一是FC占用了很大的空间,如果将第一层的FC中2048降低了512,网络结构回降低到三分之一,当然性能也随之下降(但是在tusimple上下降不大)。二是在保存网络参数的时候同事保存了state_dict和optimizer,去掉optimizer会下降一半。
可以优化后面的两个全连接,先split分开fc
您发的邮件已收到,谢谢——武龙
你好 可以加一个联系方式嘛 关于这个模型有些问题想请教一下
您发的邮件已收到,谢谢——武龙