Home
last modified time | relevance | path

Searched refs:ds_run_context (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/dataset/callback/
Dds_callback.py48 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/
Dtest_callbacks.py38 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 …]