请问一下,如果显卡的显存不是很大,batch_size只能设置到4,需要采取什么措施吗?

来源:13-11 课程总结

慕慕0439991

2021-01-03

尊敬的Kuan老师,您好,请问一下,如果显卡的显存不是很大,batch_size只能设置到4,需要采取什么措施吗?

期待您的回复!

写回答

1回答

会写代码的好厨师

2021-01-04

减小backbone的尺寸或者换掉backbone用轻量型网络结构。

0
3
慕慕0439991
回复
会写代码的好厨师
好的,谢谢老师!
2021-01-11
共3条回复

深度学习之目标检测常用算法原理+实践精讲

从原理到场景实战,掌握目标检测核心技术

878 学习 · 221 问题

查看课程