Searched refs:config_ck (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/nn/probability/toolbox/ |
| D | uncertainty_evaluation.py | 134 config_ck = CheckpointConfig(
138 config=config_ck)
187 config_ck = CheckpointConfig(keep_checkpoint_max=self.epochs)
190 config=config_ck)
|
| /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) variable
141 … ckpoint_cb = ModelCheckpoint(prefix="train_resnet_cifar10", directory="./", config=config_ck)
|
| D | resnet_cifar_normal.py | 140 config_ck = CheckpointConfig(save_checkpoint_steps=batch_num * 5, keep_checkpoint_max=10) variable
141 … ckpoint_cb = ModelCheckpoint(prefix="train_resnet_cifar10", directory="./", config=config_ck)
|
| /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) variable
142 … ckpoint_cb = ModelCheckpoint(prefix="train_resnet_cifar10", directory="./", config=config_ck)
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 365 … config_ck = CheckpointConfig(save_checkpoint_steps=self.train_config.save_checkpoint_steps,
369 config=config_ck)
|