前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
Created by: Ethantequila
在训练detection demo的时候显存达到10G,想问一下有什么办法,可以减小训练显存吗?看到fl_tarin.py里面的batch_size已经是1了,不知道还有没有其他参数可以调整。