From cf36a05e81663706096e09ecd02d712bb2005f3e Mon Sep 17 00:00:00 2001 From: caozhou Date: Wed, 9 Dec 2020 07:55:17 +0800 Subject: [PATCH] add to api document --- mindspore/train/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mindspore/train/__init__.py b/mindspore/train/__init__.py index c02107cd30..b6aff9b773 100644 --- a/mindspore/train/__init__.py +++ b/mindspore/train/__init__.py @@ -23,8 +23,9 @@ from . import amp from .amp import build_train_network from .loss_scale_manager import LossScaleManager, FixedLossScaleManager, DynamicLossScaleManager from .serialization import save_checkpoint, load_checkpoint, load_param_into_net, export, parse_print,\ - build_searched_strategy, merge_sliced_parameter + build_searched_strategy, merge_sliced_parameter, load_distributed_checkpoint __all__ = ["Model", "DatasetHelper", "amp", "connect_network_with_dataset", "build_train_network", "LossScaleManager", "FixedLossScaleManager", "DynamicLossScaleManager", "save_checkpoint", "load_checkpoint", - "load_param_into_net", "export", "parse_print", "build_searched_strategy", "merge_sliced_parameter"] + "load_param_into_net", "export", "parse_print", "build_searched_strategy", "merge_sliced_parameter", + "load_distributed_checkpoint"]