| /third_party/mindspore/mindspore/train/callback/ |
| D | _summary_collector.py | 398 def _get_tensor_collect_range(self, cb_params, dataset_sink_mode): argument
416 def _collect_at_step_end(self, cb_params, plugin_filter): argument
437 def _check_callbacks(self, cb_params): argument
475 def _collect_input_data(self, cb_params): argument
490 def _is_allowed_to_collect_input_data(self, cb_params): argument
516 def _collect_dataset_graph(self, cb_params): argument
531 def _collect_graphs(self, cb_params): argument
545 def _collect_metric(self, cb_params): argument
560 def _get_loss(self, cb_params): argument
597 def _get_optimizer(self, cb_params): argument
[all …]
|
| D | _checkpoint.py | 380 def _check_save_ckpt(self, cb_params, force_to_save): argument
394 def _save_ckpt(self, cb_params, force_to_save=False): argument
441 def _flush_from_cache(self, cb_params): 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 …def _train_ascend_sink_step(self, cb_params, train_dataset, iter_first_order, inputs, list_callbac… argument
190 …def _train_dataset_sink_process(self, epoch, train_dataset, list_callback=None, cb_params=None, si… argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | model_thor.py | 410 def _train_dataset_sink_process(self, epoch, train_dataset, list_callback=None, cb_params=None): argument
475 def _train_process(self, epoch, train_dataset, list_callback=None, cb_params=None): argument
588 def _eval_dataset_sink_process(self, valid_dataset, list_callback=None, cb_params=None): argument
626 def _eval_process(self, valid_dataset, list_callback=None, cb_params=None): argument
|
| /third_party/mindspore/mindspore/train/ |
| D | model.py | 517 …def _train_dataset_sink_process(self, epoch, train_dataset, list_callback=None, cb_params=None, si… argument
586 def _train_process(self, epoch, train_dataset, list_callback=None, cb_params=None): argument
765 def _eval_dataset_sink_process(self, valid_dataset, list_callback=None, cb_params=None): argument
802 def _eval_process(self, valid_dataset, list_callback=None, cb_params=None): argument
1084 def _flush_from_cache(self, cb_params): argument
|