Home
last modified time | relevance | path

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.py64 on_train_step_end=None, on_train_begin=None, on_train_end=None, argument
D_time_monitor.py80 def on_train_step_end(self, run_context): member in TimeMonitor
D_on_request_exit.py99 def on_train_step_end(self, run_context): member in OnRequestExit
D_mindio_ttp.py305 def on_train_step_end(self, run_context): member in MindIOTTPAdapter
D_callback.py272 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/
Dtraining_checker.py113 def on_train_step_end(self, run_context): member in TrainingChecker
/third_party/mindspore/mindspore-src/source/tests/st/optimizer/
Dcame_optimizer_shard.py53 def on_train_step_end(self, run_context): member in ModelCallback
/third_party/mindspore/mindspore-src/source/tests/st/profiler/
Dtest_step_profiler.py47 def on_train_step_end(self, run_context): member in StopAtStep
Drun_net_with_dynamic_profiler.py158 def on_train_step_end(self, run_context): member in DynamicProfiler