Searched refs:ds_run_context (Results 1 – 2 of 2) sorted by relevance
/third_party/mindspore/mindspore/dataset/callback/ |
D | ds_callback.py | 48 def ds_begin(self, ds_run_context): argument 56 def ds_epoch_begin(self, ds_run_context): argument 64 def ds_epoch_end(self, ds_run_context): argument 72 def ds_step_begin(self, ds_run_context): argument 80 def ds_step_end(self, ds_run_context): argument 155 def sync_epoch_begin(self, train_run_context, ds_run_context): argument 164 def sync_step_begin(self, train_run_context, ds_run_context): argument 183 def ds_epoch_begin(self, ds_run_context): argument 190 if ds_run_context.cur_epoch_num > 1: 197 self.sync_epoch_begin(self.epoch_run_context, ds_run_context) [all …]
|
/third_party/mindspore/tests/ut/python/dataset/ |
D | test_callbacks.py | 38 def append(self, event_name, ds_run_context): argument 39 event = [event_name, ds_run_context.cur_epoch_num, 40 ds_run_context.cur_step_num_in_epoch, ds_run_context.cur_step_num] 54 def ds_begin(self, ds_run_context): argument 55 self.append("begin", ds_run_context) 59 def ds_epoch_begin(self, ds_run_context): argument 60 self.append("epoch_begin", ds_run_context) 64 def ds_epoch_end(self, ds_run_context): argument 65 self.append("epoch_end", ds_run_context) 69 def ds_step_begin(self, ds_run_context): argument [all …]
|