megatron.global_vars.update_num_microbatches#

megatron.global_vars.update_num_microbatches(consumed_samples, consistency_check=True)#