Home
last modified time | relevance | path

Searched refs:save_checkpoint (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/
Dconvert_checkpoint.py18 from mindspore.train.serialization import load_checkpoint, save_checkpoint
62 save_checkpoint(parameter_list, "resnet_backbone.ckpt")
/third_party/mindspore/tests/ut/python/utils/
Dtest_serialize.py35 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__.py25 from .serialization import save_checkpoint, load_checkpoint, load_param_into_net, export, load, par…
Dserialization.py198 def save_checkpoint(save_obj, ckpt_file_name, integrated_save=True, function
/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/
Dconfig.py57 save_checkpoint = True variable in TrainConfig
Ddeepfm.py364 if self.train_config.save_checkpoint:
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/
Dtest_fl_fasterrcnn.py29 from mindspore.train.serialization import save_checkpoint
214 save_checkpoint(net, ckpt_path)
/third_party/mindspore/tests/st/fl/albert/
Dcloud_train.py23 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/
Dtest_gradient_accumulation.py19 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.py27 from mindspore.train.serialization import save_checkpoint, _save_graph
436save_checkpoint(network, cur_file, self._config.integrated_save, self._config.async_save,
/third_party/mindspore/tests/st/fl/cross_silo_femnist/
Dtest_cross_silo_femnist.py30 from mindspore.train.serialization import save_checkpoint
342 save_checkpoint(network, ckpt_name)
/third_party/mindspore/
DRELEASE.md2509 - Modification of save_checkpoint([!5482](https://gitee.com/mindspore/mindspore/pulls/5482))