Searched defs:on_train_step_end (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/callback/ |
| D | _lambda_callback.py | 64 on_train_step_end=None, on_train_begin=None, on_train_end=None, argument
|
| D | _time_monitor.py | 80 def on_train_step_end(self, run_context): member in TimeMonitor
|
| D | _on_request_exit.py | 99 def on_train_step_end(self, run_context): member in OnRequestExit
|
| D | _mindio_ttp.py | 305 def on_train_step_end(self, run_context): member in MindIOTTPAdapter
|
| D | _callback.py | 272 def on_train_step_end(self, run_context): member in Callback
434 def on_train_step_end(self, run_context): member in CallbackManager
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/llama/ |
| D | training_checker.py | 113 def on_train_step_end(self, run_context): member in TrainingChecker
|
| /third_party/mindspore/mindspore-src/source/tests/st/optimizer/ |
| D | came_optimizer_shard.py | 53 def on_train_step_end(self, run_context): member in ModelCallback
|
| /third_party/mindspore/mindspore-src/source/tests/st/profiler/ |
| D | test_step_profiler.py | 47 def on_train_step_end(self, run_context): member in StopAtStep
|
| D | run_net_with_dynamic_profiler.py | 158 def on_train_step_end(self, run_context): member in DynamicProfiler
|