PR #31530 Removed torch.cuda.empty_cache from train loop.

Removed torch.cuda.empty_cache from train loop. #31530

amyeroberts merged 1 commit into huggingface:main from FoamoftheSea:remove-train-loop-cache-clear

Removed torch.cuda.empty_cache from train loop.

d1884ff9

SunMarc approved these changes on 2024-06-21

SunMarc requested a review from

muellerzr 1 year ago

SunMarc requested a review from

amyeroberts 1 year ago

muellerzr approved these changes on 2024-06-21

amyeroberts approved these changes on 2024-06-21

amyeroberts merged 8b7cd402 into main 1 year ago

Reviewers

SunMarc

amyeroberts

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone