| /third_party/mindspore/mindspore/train/callback/ |
| D | _callback.py | 113 def begin(self, run_context): argument
121 def epoch_begin(self, run_context): argument
129 def epoch_end(self, run_context): argument
137 def step_begin(self, run_context): argument
145 def step_end(self, run_context): argument
153 def end(self, run_context): argument
201 def begin(self, run_context): argument
206 def epoch_begin(self, run_context): argument
211 def epoch_end(self, run_context): argument
216 def step_begin(self, run_context): argument
[all …]
|
| D | _time_monitor.py | 40 def epoch_begin(self, run_context): argument
49 def epoch_end(self, run_context): argument
|
| D | _loss_monitor.py | 46 def step_end(self, run_context): argument
|
| D | _lr_scheduler_callback.py | 60 def step_end(self, run_context): argument
|
| D | _summary_collector.py | 356 def begin(self, run_context): argument
367 def step_end(self, run_context): argument
422 def epoch_end(self, run_context): argument
425 def end(self, run_context): argument
|
| D | _checkpoint.py | 337 def step_end(self, run_context): argument
361 def end(self, run_context): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | callback.py | 42 def epoch_end(self, run_context): argument
71 def step_end(self, run_context): argument
96 def epoch_begin(self, run_context): argument
99 def epoch_end(self, run_context): argument
105 def step_begin(self, run_context): argument
108 def step_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/quantization/resnet50_quant/ |
| D | utils.py | 49 def epoch_begin(self, run_context): argument
53 def epoch_end(self, run_context): argument
63 def step_begin(self, run_context): argument
66 def step_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/quantization/mobilenetv2_quant/ |
| D | utils.py | 50 def epoch_begin(self, run_context): argument
54 def epoch_end(self, run_context): argument
64 def step_begin(self, run_context): argument
67 def step_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/ |
| D | test_yolov3.py | 65 def step_end(self, run_context): argument
77 def epoch_begin(self, run_context): argument
80 def epoch_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/ |
| D | test_yolov3_darknet53.py | 68 def step_end(self, run_context): argument
82 def epoch_begin(self, run_context): argument
85 def epoch_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/transformer/ |
| D | test_transformer.py | 122 def step_end(self, run_context): argument
139 def epoch_begin(self, run_context): argument
142 def epoch_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/ |
| D | test_deeplabv3.py | 55 def epoch_begin(self, run_context): argument
57 def step_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | callbacks.py | 51 def step_end(self, run_context): argument
105 def epoch_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/networks/models/bert/bert_performance/ |
| D | test_bert_thor.py | 84 def epoch_begin(self, run_context): argument
87 def epoch_end(self, run_context): argument
101 def epoch_end(self, run_context): argument
|
| D | test_bert_tdt_lossscale.py | 151 def step_end(self, run_context): argument
168 def epoch_begin(self, run_context): argument
171 def epoch_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/networks/models/bert/bert_precision/ |
| D | test_bert_tdt_lossscale.py | 151 def step_end(self, run_context): argument
168 def epoch_begin(self, run_context): argument
171 def epoch_end(self, run_context): argument
|
| /third_party/mindspore/mindspore/dataset/callback/ |
| D | ds_callback.py | 173 def epoch_end(self, run_context): argument
199 def step_end(self, run_context): argument
248 def end(self, run_context): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/ |
| D | test_resnet50_imagenet.py | 99 def step_end(self, run_context): argument
118 def epoch_begin(self, run_context): argument
121 def epoch_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/deeplabv3/ |
| D | train_one_epoch_with_loss.py | 51 def epoch_begin(self, run_context): argument
53 def step_end(self, run_context): argument
|
| /third_party/mindspore/mindspore/train/train_thor/ |
| D | model_thor.py | 144 def _train_gpu_sink_step(self, cb_params, inputs, list_callback, iter_first_order, run_context): argument
169 …nd_sink_step(self, cb_params, train_dataset, iter_first_order, inputs, list_callback, run_context): argument
|
| /third_party/mindspore/tests/st/fl/cross_silo_femnist/ |
| D | test_cross_silo_femnist.py | 188 def step_end(self, run_context): argument
208 def epoch_begin(self, run_context): argument
211 def epoch_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_resnet50_ascend.py | 56 def epoch_begin(self, run_context): argument
59 def epoch_end(self, run_context): argument
64 def step_begin(self, run_context): argument
67 def step_end(self, run_context): argument
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | test_resnet_cifar_1p.py | 110 def step_end(self, run_context): argument
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_callbacks.py | 238 def epoch_end(self, run_context): argument
243 def step_end(self, run_context): argument
|