mindformers.core.callback.ProfileMonitor

class mindformers.core.callback.ProfileMonitor(start_step=1, stop_step=10, output_path=None, start_profile=True, profile_communication=False, profile_memory=True, **kwargs)

Profile analysis in training.

step_begin(run_context)

Start profile at the begin of step.

Args:

run_context (RunContext): Context of the train running.

step_end(run_context)

Stop profile at the end of step.

Args:

run_context (RunContext): Context of the train running.