Skip to content

fix: remove all buffers registered besides the supported ops#199

Open
six-m wants to merge 1 commit into
Lyken17:masterfrom
six-m:fix-state-persistence
Open

fix: remove all buffers registered besides the supported ops#199
six-m wants to merge 1 commit into
Lyken17:masterfrom
six-m:fix-state-persistence

Conversation

@six-m
Copy link
Copy Markdown

@six-m six-m commented Mar 9, 2023

The buffer named 'total_ops' and 'total_params' maybe remain inside the model after calling profile function when the submodule is not supported. And the checkpoints exported contains the buffers which affect loading.

This PR fix this.

@six-m six-m changed the title fix: remove all buffers registered beside the supported ops fix: remove all buffers registered besides the supported ops Mar 9, 2023
@six-m six-m force-pushed the fix-state-persistence branch from ae677a0 to 201197f Compare March 9, 2023 03:13
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