PR #35440 Release GPU memory after Optuna trial

Release GPU memory after Optuna trial #35440

ArthurZucker merged 4 commits into huggingface:main from JanetVictorious:feature/memory-release-for-hpo-optuna

Release GPU memory after trial

58ecbdd0

Merge remote-tracking branch 'upstream/main' into feature/memory-rele…

386ebf78

SunMarc approved these changes on 2024-12-30

SunMarc requested a review from

ArthurZucker 356 days ago

Merge remote-tracking branch 'upstream/main' into feature/memory-rele…

0312981d

Update to use release_memory from accelerate.utils.memory after sugge…

26bba18a

ArthurZucker approved these changes on 2025-01-07

ArthurZucker merged 628cd838 into main 348 days ago

JanetVictorious deleted the feature/memory-release-for-hpo-optuna branch 348 days ago

Reviewers

ArthurZucker

SunMarc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone