yzqxy / Yolov8_obb_Prune_Track

GNU General Public License v3.0
181 stars 14 forks source link

训练过程中每3个epoch会卡顿变慢 #25

Open jiwenfei opened 11 months ago

jiwenfei commented 11 months ago

显卡是3090ti 24g,cpu 是 intel xeon 主频2.2ghz,20个内核 网络输入图是480,batch-size 256,各种数据增强,workers=16,18,20,24 都试过 gpu占用 22个g,cpu占80-100%。 出现的现象:每训3个epoch大约会卡3-5分钟,这3个epoch会很快。 降低batchsize也会出现这样的问题 想请教一下大佬,这个该往哪个方向调整