megatron.training.train_step

megatron.training.train_step(forward_step_func, data_iterator, model, optimizer, opt_param_scheduler, iteration, args)

Single training step.