Searched refs:save_checkpoint (Results 1 – 12 of 12) sorted by relevance
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/ |
D | convert_checkpoint.py | 18 from mindspore.train.serialization import load_checkpoint, save_checkpoint 62 save_checkpoint(parameter_list, "resnet_backbone.ckpt")
|
/third_party/mindspore/tests/ut/python/utils/ |
D | test_serialize.py | 35 from mindspore.train.serialization import save_checkpoint, load_checkpoint, load_param_into_net, \ 122 save_checkpoint(parameter_list, ckpt_file_name) 146 save_checkpoint(parameter_list, ckpt_file_name, append_dict=append_dict) 318 save_checkpoint(train_network, ckpt_file_name="./new_ckpt.ckpt") 342 save_checkpoint(train_network, ckpt_file_name=ckpt_path, enc_key=key, enc_mode=mode) 346 save_checkpoint(train_network, ckpt_file_name=ckpt_path, enc_key=key, enc_mode=mode)
|
/third_party/mindspore/mindspore/train/ |
D | __init__.py | 25 from .serialization import save_checkpoint, load_checkpoint, load_param_into_net, export, load, par…
|
D | serialization.py | 198 def save_checkpoint(save_obj, ckpt_file_name, integrated_save=True, function
|
/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
D | config.py | 57 save_checkpoint = True variable in TrainConfig
|
D | deepfm.py | 364 if self.train_config.save_checkpoint:
|
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/ |
D | test_fl_fasterrcnn.py | 29 from mindspore.train.serialization import save_checkpoint 214 save_checkpoint(net, ckpt_path)
|
/third_party/mindspore/tests/st/fl/albert/ |
D | cloud_train.py | 23 from mindspore.train.serialization import save_checkpoint, load_checkpoint 205 save_checkpoint(server_network_train_cell.network, output_path)
|
/third_party/mindspore/tests/st/networks/ |
D | test_gradient_accumulation.py | 19 from mindspore.train.serialization import save_checkpoint 166 save_checkpoint(self._train_forward_backward, "gradient_accumulation.ckpt",)
|
/third_party/mindspore/mindspore/train/callback/ |
D | _checkpoint.py | 27 from mindspore.train.serialization import save_checkpoint, _save_graph 436 … save_checkpoint(network, cur_file, self._config.integrated_save, self._config.async_save,
|
/third_party/mindspore/tests/st/fl/cross_silo_femnist/ |
D | test_cross_silo_femnist.py | 30 from mindspore.train.serialization import save_checkpoint 342 save_checkpoint(network, ckpt_name)
|
/third_party/mindspore/ |
D | RELEASE.md | 2509 - Modification of save_checkpoint([!5482](https://gitee.com/mindspore/mindspore/pulls/5482))
|