Searched refs:keep_checkpoint_max (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/tests/ut/python/train/ |
| D | test_run_config.py | 24 keep_checkpoint_max = 5
27 keep_checkpoint_max)
30 assert config.keep_checkpoint_max == keep_checkpoint_max
32 assert policy['keep_checkpoint_max'] == keep_checkpoint_max
40 assert config.keep_checkpoint_max == 5
48 CheckpointConfig(keep_checkpoint_max='abc')
57 CheckpointConfig(keep_checkpoint_max=-1)
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _checkpoint.py | 140 keep_checkpoint_max=5, argument
153 if keep_checkpoint_max is not None:
154 keep_checkpoint_max = Validator.check_non_negative_int(keep_checkpoint_max)
163 not keep_checkpoint_max and not keep_checkpoint_per_n_minutes:
172 self._keep_checkpoint_max = keep_checkpoint_max
198 def keep_checkpoint_max(self): member in CheckpointConfig
241 'keep_checkpoint_max': self.keep_checkpoint_max,
412 …if self._config.keep_checkpoint_max and 0 < self._config.keep_checkpoint_max <= self._manager.ckpo…
|
| /third_party/mindspore/tests/ut/python/utils/ |
| D | test_callback.py | 96 keep_checkpoint_max=5,
198 keep_checkpoint_max=5,
227 keep_checkpoint_max=0,
256 keep_checkpoint_max=5,
372 keep_checkpoint_max=5,
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | config.py | 60 keep_checkpoint_max = 15 variable in TrainConfig
|
| D | deepfm.py | 366 keep_checkpoint_max=self.train_config.keep_checkpoint_max)
|
| /third_party/mindspore/tests/st/quantization/lenet_quant/ |
| D | test_lenet_quant.py | 78 keep_checkpoint_max=cfg.keep_checkpoint_max)
|
| /third_party/mindspore/mindspore/nn/probability/toolbox/ |
| D | uncertainty_evaluation.py | 135 keep_checkpoint_max=self.epochs)
187 config_ck = CheckpointConfig(keep_checkpoint_max=self.epochs)
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/ |
| D | train_and_test_multinpu_ci_data_parallel.py | 96 …config = CheckpointConfig(save_checkpoint_steps=ds_train.get_dataset_size(), keep_checkpoint_max=5)
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_parameter_merge.py | 88 ckpt_config = CheckpointConfig(keep_checkpoint_max=1)
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet_cifar_memreuse.py | 140 config_ck = CheckpointConfig(save_checkpoint_steps=batch_num * 5, keep_checkpoint_max=10)
|
| D | resnet_cifar_normal.py | 140 config_ck = CheckpointConfig(save_checkpoint_steps=batch_num * 5, keep_checkpoint_max=10)
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | resnet_cifar.py | 141 config_ck = CheckpointConfig(save_checkpoint_steps=batch_num * 5, keep_checkpoint_max=10)
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | optimizer_parallel.py | 262 ckpt_config = CheckpointConfig(keep_checkpoint_max=1)
|
| D | multifieldembeddinglookup_parallel.py | 252 ckpt_config = CheckpointConfig(keep_checkpoint_max=1)
|
| D | parallel_strategy_search.py | 287 ckpt_config = CheckpointConfig(keep_checkpoint_max=1)
|