THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection

Home Page:https://arxiv.org/abs/2405.14458

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The usage of graphics card memory increases with the number of training epochs

ayitime opened this issue · comments

Hi, I ran the YOLO v10 code but found an issue where the memory usage has been increasing from 3GB to 9GB, and there is no sign of it stopping. What is the reason for this
1721216411647

I run into the exact same issue. Did you find a solution? Thanks!

Also I believe Issue #361 is the same issue.