megatron.global_vars.get_current_global_batch_size#

megatron.global_vars.get_current_global_batch_size()#