Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Patch] Add a clear_cache_freq option in the commandline. #63

Merged
merged 2 commits into from
Sep 3, 2024

Conversation

garrett4wade
Copy link
Contributor

This option enables us to manually control the frequency of cleaning CUDA cache during training, avoiding the OOM issue.

@garrett4wade garrett4wade merged commit c68c8d6 into main Sep 3, 2024
3 checks passed
@garrett4wade garrett4wade deleted the clear-cache-option branch September 3, 2024 07:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant