megatron.global_vars.get_current_global_batch_size# megatron.global_vars.get_current_global_batch_size()#