Searched refs:save_checkpoint_steps (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/tests/ut/python/train/ |
| D | test_run_config.py | 23 save_checkpoint_steps = 1
26 config = CheckpointConfig(save_checkpoint_steps,
29 assert config.save_checkpoint_steps == save_checkpoint_steps
38 assert config.save_checkpoint_steps == 1
44 CheckpointConfig(save_checkpoint_steps='abc')
53 CheckpointConfig(save_checkpoint_steps=-1)
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _checkpoint.py | 138 save_checkpoint_steps=1, argument
149 if save_checkpoint_steps is not None:
150 save_checkpoint_steps = Validator.check_non_negative_int(save_checkpoint_steps)
162 if not save_checkpoint_steps and not save_checkpoint_seconds and \
167 self._save_checkpoint_steps = save_checkpoint_steps
188 def save_checkpoint_steps(self): member in CheckpointConfig
239 checkpoint_policy = {'save_checkpoint_steps': self.save_checkpoint_steps,
382 if self._config.save_checkpoint_steps and self._config.save_checkpoint_steps > 0:
383 … if cb_params.cur_step_num >= self._last_triggered_step + self._config.save_checkpoint_steps \
|
| /third_party/mindspore/tests/ut/python/utils/ |
| D | test_callback.py | 94 save_checkpoint_steps=16,
196 save_checkpoint_steps=16,
225 save_checkpoint_steps=16,
254 save_checkpoint_steps=16,
370 save_checkpoint_steps=16,
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | config.py | 59 save_checkpoint_steps = 1 variable in TrainConfig
|
| D | deepfm.py | 365 … config_ck = CheckpointConfig(save_checkpoint_steps=self.train_config.save_checkpoint_steps,
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/ |
| D | train_and_test_multinpu_ci_data_parallel.py | 96 …ckptconfig = CheckpointConfig(save_checkpoint_steps=ds_train.get_dataset_size(), keep_checkpoint_m…
|
| /third_party/mindspore/tests/st/quantization/lenet_quant/ |
| D | test_lenet_quant.py | 77 config_ckpt = CheckpointConfig(save_checkpoint_steps=cfg.epoch_size * step_size,
|
| /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)
|