data icon indicating copy to clipboard operation
data copied to clipboard

Data loader directly use loaded state dict when save right after load

Open MiiiraK opened this issue 1 year ago • 1 comments

Summary: If no iterator is created in the middle of load_state_dict and state_dict calls, we should be able to directly return the original state dict without triggering reading service because the states are unchanged

Reviewed By: xunnanxu

Differential Revision: D54102267

MiiiraK avatar Feb 26 '24 18:02 MiiiraK

This pull request was exported from Phabricator. Differential Revision: D54102267

facebook-github-bot avatar Feb 26 '24 18:02 facebook-github-bot

PR already merged in the main branch - https://github.com/pytorch/data/commit/39b3497517d5850c600c6b18e73cf5a346a539bf

gokulavasan avatar Mar 22 '24 22:03 gokulavasan