首页 » 技术SOS » 通信技术 » 其他 » 同样一张2080Ti,跑30G的bisenet轻轻松松,反而跑十几G的更轻量化的网络就跑不动了?

同样一张2080Ti,跑30G的bisenet轻轻松松,反而跑十几G的更轻量化的网络就跑不动了?

菜鸟
2022-09-20 11:21:34     悬赏5分

同样一张2080Ti,跑30G的bisenet轻轻松松,反而跑十几G的更轻量化的网络就跑不动了,会不会因为深度可分离卷积和非对称卷积这些pytorch没有优化呢?  比如我一张2080ti,512x1024下bisenetv2可以bs=4,反而跑今年TMM一篇flops只有十G的FBSnet 两张2080ti都跑不起?


分享
关键词: 网络, 就跑  
菜鸟
2022-09-20 11:28:59
1楼

和分辨率有关吧,太大的输入size也会,再不行就把奇奇怪怪的增广关了,比如muti scale这种。build model之后,rand一个nchw的tensor进去看看,没有溢出就是训练框架的设置问题。

专家
2022-09-20 23:02:48
2楼

换3.080跑跑

高工
2022-09-21 20:13:03
3楼

没有显卡跑过

高工
2022-09-25 15:12:24
4楼

2080TI?

专家
2022-10-09 19:01:54
5楼

位宽多少?