Aurxs

@Y-T-G Thanks for the suggestion. I've run the provided code snippet to test the `yolo11n-cls.pt` model. I've run the test with `yolo11n-cls.pt` as you proposed. The results are quite interesting....

> So it grows continuously for detection model training, but not for classification model training? Based on the results, yes.

@glenn-jocher Thanks for the clear instructions. I have run the test as you proposed. But the memory increase problem persists. Here's the result below: Disable heavy augs and re-test on...

I tried to use github copilot agent to fix this problem.He added a memory reclamation strategy for MPS training.And it actually works in the test.Perhaps merging PR could be considered...

@glenn-jocher Hello! I open a PR to solve the problem yesterday. See https://github.com/ultralytics/ultralytics/pull/22695. Please review my commit to merge the PR and fix the issue. Thanks again for your help!