Implement save_state_dict
and load_state_dict
in serialization module
#2065
Milestone
save_state_dict
and load_state_dict
in serialization module
#2065
Follow-up issue after #1938 that introduced
split_state_dict_into_shards
. Let's go one step further and implementsave_state_dict
as described in #1938.(once that's done, we will be able to reuse it in
PyTorchMixinHub
as well)The text was updated successfully, but these errors were encountered: