Skip to content

Conversation

@rwtarpit
Copy link
Contributor

@rwtarpit rwtarpit commented Feb 6, 2026

Fixes #{issue number}
NA
Description:
Refactored Checkpoint.state_dict and Checkpoint.load_state_dict to remove hard-coded "_saved" strings.
Uses base class Serializable's validation mechanism to verify state_dict.
Check list:

  • New tests are added (if a new feature is added)
  • New doc strings: description and/or example code are in RST format
  • Documentation is updated (if required)

@github-actions github-actions bot added the module: handlers Core Handlers module label Feb 6, 2026
@rwtarpit rwtarpit changed the title refactor: improve Checkpoint serialization consistency for state_dict refactor: improve Checkpoint-Serialize consistency for state_dict Feb 6, 2026
Copy link
Collaborator

@vfdev-5 vfdev-5 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @rwtarpit !

@vfdev-5 vfdev-5 enabled auto-merge February 9, 2026 11:32
@vfdev-5 vfdev-5 added this pull request to the merge queue Feb 9, 2026
Merged via the queue into pytorch:master with commit ef7f17e Feb 9, 2026
22 of 24 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

module: handlers Core Handlers module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants