Skip to content

fix(utils): handle None values safely in MemoryTrackerDataPoint properties - #3

Open
yappermoar-boop wants to merge 1 commit into
NousResearch:mainfrom
yappermoar-boop:fix-memory-tracker-datapoint-none-guard
Open

fix(utils): handle None values safely in MemoryTrackerDataPoint properties#3
yappermoar-boop wants to merge 1 commit into
NousResearch:mainfrom
yappermoar-boop:fix-memory-tracker-datapoint-none-guard

Conversation

@yappermoar-boop

Copy link
Copy Markdown

Updated MemoryTrackerDataPoint in nemo_rl/utils/memory_tracker.py to safely handle None values for memory_used_after_stage_gb and variables_after_stage. Previously, calling mem_used_diff_gb or new_variables on an uncompleted stage snapshot threw unhandled TypeError exceptions.

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