xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!9690 add interface to api document 

From: @caozhou_huawei
Reviewed-by: @kingxian,@zh_qh
Signed-off-by: @kingxian
This commit is contained in:
mindspore-ci-bot 2020-12-09 11:35:03 +08:00 committed by Gitee
parent 4a7c87f442 cf36a05e81
commit 90ae2a462e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/train/__init__.py
View File

@ -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"]