SiamTrackers icon indicating copy to clipboard operation
SiamTrackers copied to clipboard

您好,我在训练SiamRPN时显存利用低

Open jyCqupt opened this issue 3 years ago • 2 comments

数据集: GOT10k,cpu: AMD EPYC 7543 32-Core Processor GPU: RTX3090 cpu已经拉满了,但GPU利用率低,偶尔跳下20%,40%大部分时间为0 修改num_workers和batch_size参数无果 且每次train_loader处理完一次后都会卡顿一下

jyCqupt avatar Jun 30 '22 11:06 jyCqupt

训练的时候看一下每次读取数据的耗时,是不是卡在读取数据上了. CPU内存最好>64G, num_workers=4 or 8 or 12. 机械硬盘一般比较慢, 固态硬盘的还好. 另外, 可以尝试训练NanoTrack模型,效果可能比SiamRPN要好

HonglinChu avatar Jul 02 '22 02:07 HonglinChu

感谢大佬 应该是卡在读取数据了 我再研究一下

------------------ 原始邮件 ------------------ 发件人: @.>; 发送时间: 2022年7月2日(星期六) 上午10:04 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [HonglinChu/SiamTrackers] 您好,我在训练SiamRPN时显存利用低 (Issue #77)

训练的时候看一下每次读取数据的耗时,是不是卡在读取数据上了. CPU内存最好>64G, num_workers=4 or 8 or 12. 机械硬盘一般比较慢, 固态硬盘的还好. 另外, 可以尝试训练NanoTrack模型,效果可能比SiamRPN要好

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

jyCqupt avatar Jul 02 '22 02:07 jyCqupt