mmdet-rfla icon indicating copy to clipboard operation
mmdet-rfla copied to clipboard

Published 20 hours ago • verified publisher icon Chasel-Tsui

DOTAv2.0训练显存不足

Open yuhua666 opened this issue 1 year ago • 1 comments

我也是3090单卡24GB，我训练DOTAv2.0使用跟您相同的配置时显存不足，发现要调小img_scale才可以，请问调小这个会显著影响性能吗？

yuhua666 avatar Apr 15 '23 03:04 yuhua666

有以下方法可以减少显存，(1) 调小配置文件中的gpu_assign_thr (首推) (2) 调小batchsize，同时线性降低学习率 (3) fp16训练 (略微降低性能，但很小)

Chasel-Tsui avatar Apr 18 '23 01:04 Chasel-Tsui