From a91539805a49e2124715708096f0c7980aa18482 Mon Sep 17 00:00:00 2001 From: co63oc Date: Sun, 4 Feb 2024 10:46:16 +0800 Subject: [PATCH] Fix --- python/paddle/incubate/distributed/utils/io/dist_load.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/paddle/incubate/distributed/utils/io/dist_load.py b/python/paddle/incubate/distributed/utils/io/dist_load.py index 0d6c91e75fc6b2..621f06e3eee701 100644 --- a/python/paddle/incubate/distributed/utils/io/dist_load.py +++ b/python/paddle/incubate/distributed/utils/io/dist_load.py @@ -61,7 +61,7 @@ def load(path, **configs): model_state_dict = paddle.incubate.distributed.utils.io.load(path="path/to/load.pdparams") dist_model.set_state_dict(model_state_dict) - # load optimizer satte dict + # load optimizer state dict optimizer_state_dict = paddle.incubate.distributed.utils.io.load(path="path/to/load.pdopt") dist_optimizer.set_state_dict(optimizer_state_dict) @@ -100,7 +100,7 @@ def load_with_place(path, **configs): origin_place = paddle.get_device() paddle.set_device(place) - configs = _remove_not_supported_itmes(configs) + configs = _remove_not_supported_items(configs) state_dict = paddle.load(path, **configs) paddle.set_device(origin_place) @@ -108,7 +108,7 @@ def load_with_place(path, **configs): return state_dict -def _remove_not_supported_itmes(configs): +def _remove_not_supported_items(configs): __supported_by_load__ = [ "model_filename", "params_filename",