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.

Sign up for GitHub

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

emphasize generation cleanup in gc callback #671

Closed
wants to merge 1 commit into from
+2 −2
Closed

emphasize generation cleanup in gc callback #671

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions torchtnt/framework/callbacks/garbage_collector.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,11 +33,11 @@ class GarbageCollector(Callback):
Synchronizing the garbage collection can lead to a performance improvement.
The frequency of garbage collection must be tuned based on the application at hand.

By default, this callback does generation 1 collection every step. This can free up
By default, this callback does **generation 1** collection every step. This can free up
some objects to be reaped with minimal overhead compared to the full garbage collection.

Args:
step_interval: number of steps to run before each collection
step_interval: number of steps to run before executing a full garbage cleanup
"""

def __init__(self, step_interval: int) -> None:
Expand Down
Loading