Searched defs:on_train_step_begin (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/callback/ |
| D | _lambda_callback.py | 63 def __init__(self, on_train_epoch_begin=None, on_train_epoch_end=None, on_train_step_begin=None, argument
|
| D | _time_monitor.py | 68 def on_train_step_begin(self, run_context): member in TimeMonitor
|
| D | _callback.py | 263 def on_train_step_begin(self, run_context): member in Callback
429 def on_train_step_begin(self, run_context): member in CallbackManager
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/llama/ |
| D | training_checker.py | 108 def on_train_step_begin(self, run_context): member in TrainingChecker
|
| /third_party/mindspore/mindspore-src/source/tests/st/profiler/ |
| D | test_step_profiler.py | 41 def on_train_step_begin(self, run_context): member in StopAtStep
|
| D | run_net_with_dynamic_profiler.py | 149 def on_train_step_begin(self, run_context): member in DynamicProfiler
|