Home
last modified time | relevance | path

Searched refs:ModelCheckpoint (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/tests/ut/python/utils/
Dtest_callback.py31 from mindspore.train.callback import ModelCheckpoint, RunContext, LossMonitor, _InternalCallbackPar…
83 ModelCheckpoint(123)
84 ModelCheckpoint(directory="./")
86 ModelCheckpoint(config='type_error')
87 ModelCheckpoint(config=CheckpointConfig())
88 ModelCheckpoint(prefix="ckpt_2", directory="./test_files")
109 ckpoint_cb = ModelCheckpoint(prefix="test_ckpt", directory='./test_files', config=train_config)
200 ckpt_cb = ModelCheckpoint(config=train_config)
215 ckpt_cb2 = ModelCheckpoint(config=train_config)
229 ckpt_cb = ModelCheckpoint(config=train_config)
[all …]
/third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/
Dtrain_and_test_multinpu_ci_data_parallel.py21 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, TimeMonitor
97 ckpoint_cb = ModelCheckpoint(prefix='widedeep_train',
/third_party/mindspore/tests/ut/python/parallel/
Dtest_parameter_merge.py23 from mindspore.train.callback import CheckpointConfig, ModelCheckpoint
90 ckpt_cb = ModelCheckpoint(prefix="parallel", directory=ckpt_path, config=ckpt_config)
/third_party/mindspore/mindspore/train/callback/
D__init__.py25 from ._checkpoint import ModelCheckpoint
D_checkpoint.py281 class ModelCheckpoint(Callback): class
303 super(ModelCheckpoint, self).__init__()
D_summary_collector.py33 from mindspore.train.callback import Callback, ModelCheckpoint
871 if isinstance(callback, ModelCheckpoint):
/third_party/mindspore/mindspore/nn/probability/toolbox/
Duncertainty_evaluation.py23 from mindspore.train.callback import LossMonitor, ModelCheckpoint, CheckpointConfig
136 ckpoint_cb = ModelCheckpoint(prefix='checkpoint_epi_uncer_model',
188 ckpoint_cb = ModelCheckpoint(prefix='checkpoint_ale_uncer_model',
/third_party/mindspore/tests/st/quantization/lenet_quant/
Dtest_lenet_quant.py26 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor
79 ckpt_callback = ModelCheckpoint(prefix="ckpt_lenet_quant"+optim_option, config=config_ckpt)
/third_party/mindspore/tests/st/mem_reuse/
Dresnet_cifar_memreuse.py32 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor
141 … ckpoint_cb = ModelCheckpoint(prefix="train_resnet_cifar10", directory="./", config=config_ck)
Dresnet_cifar_normal.py32 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor
141 … ckpoint_cb = ModelCheckpoint(prefix="train_resnet_cifar10", directory="./", config=config_ck)
/third_party/mindspore/tests/st/tbe_networks/
Dresnet_cifar.py23 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor
142 … ckpoint_cb = ModelCheckpoint(prefix="train_resnet_cifar10", directory="./", config=config_ck)
/third_party/mindspore/tests/st/auto_parallel/
Doptimizer_parallel.py31 from mindspore.train.callback import ModelCheckpoint
264 ckpt_callback = ModelCheckpoint(prefix='parallel', directory=ckpt_path,
Dmultifieldembeddinglookup_parallel.py27 from mindspore.train.callback import ModelCheckpoint
254 ckpt_callback = ModelCheckpoint(prefix='parallel', directory=ckpt_path,
Dparallel_strategy_search.py30 from mindspore.train.callback import ModelCheckpoint
289 ckpt_callback = ModelCheckpoint(prefix='parallel', directory=ckpt_path,
/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/
Ddeepfm.py29 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig
367 ckpt_cb = ModelCheckpoint(prefix=self.train_config.ckpt_file_name_prefix,