search

Chasel-Tsui / mmdet-rfla

ECCV22: RFLA
MIT License
247 stars 23 forks source link

DOTAv2.0训练显存不足 #52

Open yuhua666 opened 1 year ago

yuhua666 commented 1 year ago

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

Chasel-Tsui commented 1 year ago

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