Model parameters: d_model 224 ffw_size 896 kv_size 32 n_heads 7 n_layers 4
Megatron-DeepSpeed/pretrain_gpt.py --tensor-model-parallel-size 1 --pipeline-model-parallel-size 1 --num-layers 4 --hidden-size 224 --num-attention-heads 7 --kv-channels 32 --ffn-hidden-size 896 --seq-length 2048 --max-position-embeddings 2048 --micro-batch-size 32 --global-batch-size 256 --train-samples 1 --vocab-file gpt2/vocab.json --merge-file gpt2/merges.txt --clip-grad 1.0 --kill-switch-path kill-switch-14m300b100mval --bf16 --optimizer adam --adam-beta1 0.9 --adam-beta2 0.999 --adam-eps 1e-8 --lr 2e-4 --min-lr 2e-5 --lr-decay-style cosine --lr-decay-samples 1 --lr-warmup-samples 0 --clip-grad 1.0 --weight-decay 1e-1 --override-lr-scheduler --no-load-optim --reset-progress --log-interval 10 --save-interval 1000 --eval-interval 1 --eval-iters 100 --tensorboard-dir tensorboard_14m300b100mval --tensorboard-queue-size 5 --log-timers-to-tensorboard --log-batch-size-to-tensorboard --log-validation-ppl-to-tensorboard --save checkpoints_14m300b100m --load checkpoints_14m300b100m --train-weighted-split-paths-path train100m.txt --valid-weighted-split-paths-path val.txt --data-impl mmap --deepspeed --deepspeed_config ds_configs/3425905.json --zero-stage 0
START 3425905: Fri 28 Apr 2023 12:41:20 AM EEST
0: 
0: 
0: ======================= ROCm System Management Interface =======================
0: ================================= Concise Info =================================
0: GPU  Temp   AvgPwr  SCLK    MCLK     Fan  Perf  PwrCap  VRAM%  GPU%  
0: 0    46.0c  95.0W   800Mhz  1600Mhz  0%   auto  560.0W    0%   0%    
0: 1    44.0c  N/A     800Mhz  1600Mhz  0%   auto  0.0W      0%   0%    
0: 2    42.0c  92.0W   800Mhz  1600Mhz  0%   auto  560.0W    0%   0%    
0: 3    40.0c  N/A     800Mhz  1600Mhz  0%   auto  0.0W      0%   0%    
0: 4    46.0c  81.0W   800Mhz  1600Mhz  0%   auto  560.0W    0%   0%    
0: 5    44.0c  N/A     800Mhz  1600Mhz  0%   auto  0.0W      0%   0%    
0: 6    40.0c  84.0W   800Mhz  1600Mhz  0%   auto  560.0W    0%   0%    
0: 7    43.0c  N/A     800Mhz  1600Mhz  0%   auto  0.0W      0%   0%    
0: ================================================================================
0: ============================= End of ROCm SMI Log ==============================
0: Launching on nid007289 (0/1), master nid007289 port 9999, GPUs 8, CUDA: True
0: using world size: 8, data-parallel-size: 8, tensor-model-parallel size: 1, pipeline-model-parallel size: 1 
0: accumulate and all-reduce gradients in fp32 for bfloat16 data type.
0: using torch.bfloat16 for parameters ...
0: ------------------------ arguments ------------------------
0:   abort_on_unmet_fused_kernel_constraints ......... False
0:   accumulate_allreduce_grads_in_fp32 .............. True
0:   adam_beta1 ...................................... 0.9
0:   adam_beta2 ...................................... 0.999
0:   adam_eps ........................................ 1e-08
0:   adlr_autoresume ................................. False
0:   adlr_autoresume_interval ........................ 1000
0:   apply_query_key_layer_scaling ................... True
0:   apply_residual_connection_post_layernorm ........ False
0:   attention_dropout ............................... 0.1
0:   attention_softmax_in_fp32 ....................... False
0:   bert_binary_head ................................ True
0:   bert_load ....................................... None
0:   bf16 ............................................ True
0:   bias_dropout_fusion ............................. True
0:   bias_gelu_fusion ................................ True
0:   biencoder_projection_dim ........................ 0
0:   biencoder_shared_query_context_model ............ False
0:   block_data_path ................................. None
0:   checkpoint_activations .......................... False
0:   checkpoint_in_cpu ............................... False
0:   checkpoint_num_layers ........................... 1
0:   clip_grad ....................................... 1.0
0:   codecarbon_dir .................................. None
0:   consumed_train_samples .......................... 0
0:   consumed_train_tokens ........................... 0
0:   consumed_valid_samples .......................... 0
0:   contigious_checkpointing ........................ False
0:   cpu_optimizer ................................... False
0:   cpu_torch_adam .................................. False
0:   curriculum_learning ............................. False
0:   data_impl ....................................... mmap
0:   data_parallel_size .............................. 8
0:   data_path ....................................... None
0:   dataloader_type ................................. single
0:   DDP_impl ........................................ local
0:   decoder_seq_length .............................. None
0:   deepscale ....................................... False
0:   deepscale_config ................................ None
0:   deepspeed ....................................... True
0:   deepspeed_activation_checkpointing .............. False
0:   deepspeed_config ................................ ds_configs/3425905.json
0:   deepspeed_mpi ................................... False
0:   distribute_checkpointed_activations ............. False
0:   distributed_backend ............................. nccl
0:   embed_layernorm ................................. False
0:   embedding_path .................................. None
0:   encoder_seq_length .............................. 2048
0:   eod_mask_loss ................................... False
0:   eval_interval ................................... 1
0:   eval_iters ...................................... 100
0:   eval_only ....................................... None
0:   evidence_data_path .............................. None
0:   exit_duration_in_mins ........................... None
0:   exit_interval ................................... None
0:   ffn_hidden_size ................................. 896
0:   finetune ........................................ False
0:   fp16 ............................................ False
0:   fp16_lm_cross_entropy ........................... False
0:   fp32_residual_connection ........................ False
0:   gigaflos_no_embeds .............................. 0
0:   global_batch_size ............................... 256
0:   glu_activation .................................. None
0:   hidden_dropout .................................. 0.1
0:   hidden_size ..................................... 224
0:   hysteresis ...................................... 2
0:   ict_head_size ................................... None
0:   ict_load ........................................ None
0:   img_dim ......................................... 224
0:   indexer_batch_size .............................. 128
0:   indexer_log_interval ............................ 1000
0:   inference ....................................... False
0:   init_method_std ................................. 0.02
0:   init_method_xavier_uniform ...................... False
0:   initial_loss_scale .............................. 4294967296
0:   kill_switch_path ................................ kill-switch-14m300b100mval
0:   kv_channels ..................................... 32
0:   layer_norm_fusion ............................... True
0:   layernorm_epsilon ............................... 1e-05
0:   lazy_mpu_init ................................... None
0:   load ............................................ checkpoints_14m300b100m
0:   local_rank ...................................... None
0:   log_batch_size_to_tensorboard ................... True
0:   log_interval .................................... 10
0:   log_learning_rate_to_tensorboard ................ True
0:   log_level ....................................... None
0:   log_level_replica ............................... None
0:   log_loss_scale_to_tensorboard ................... True
0:   log_num_zeros_in_grad ........................... False
0:   log_params_norm ................................. False
0:   log_path ........................................ None
0:   log_timers_to_tensorboard ....................... True
0:   log_validation_ppl_to_tensorboard ............... True
0:   loss_on_targets_only ............................ False
0:   loss_scale ...................................... None
0:   loss_scale_window ............................... 1000
0:   lr .............................................. 0.0002
0:   lr_decay_iters .................................. None
0:   lr_decay_samples ................................ 1
0:   lr_decay_style .................................. cosine
0:   lr_decay_tokens ................................. None
0:   lr_warmup_fraction .............................. None
0:   lr_warmup_iters ................................. 0
0:   lr_warmup_samples ............................... 0
0:   make_vocab_size_divisible_by .................... 128
0:   mask_prob ....................................... 0.15
0:   masked_softmax_fusion ........................... True
0:   max_position_embeddings ......................... 2048
0:   mean_noise_span_length .......................... None
0:   memory_centric_tiled_linear ..................... False
0:   merge_file ...................................... gpt2/merges.txt
0:   micro_batch_size ................................ 32
0:   min_loss_scale .................................. 1.0
0:   min_lr .......................................... 2e-05
0:   mmap_warmup ..................................... False
0:   no_load_optim ................................... True
0:   no_load_rng ..................................... None
0:   no_save_optim ................................... None
0:   no_save_rng ..................................... None
0:   noise_density ................................... None
0:   num_attention_heads ............................. 7
0:   num_channels .................................... 3
0:   num_classes ..................................... 1000
0:   num_layers ...................................... 4
0:   num_layers_per_virtual_pipeline_stage ........... None
0:   num_workers ..................................... 2
0:   onnx_safe ....................................... None
0:   openai_gelu ..................................... False
0:   optimizer ....................................... adam
0:   optimizer_fusion ................................ True
0:   override_lr_scheduler ........................... True
0:   pad_vocab_size_to ............................... None
0:   params_dtype .................................... torch.bfloat16
0:   partition_activations ........................... False
0:   patch_dim ....................................... 16
0:   pipeline_model_parallel_size .................... 1
0:   position_embedding_type ......................... PositionEmbeddingType.absolute
0:   pp_partition_method ............................. None
0:   profile_backward ................................ False
0:   query_in_block_prob ............................. 0.1
0:   rampup_batch_size ............................... None
0:   rank ............................................ 0
0:   remote_device ................................... none
0:   reset_attention_mask ............................ False
0:   reset_position_ids .............................. False
0:   reset_progress .................................. True
0:   retriever_report_topk_accuracies ................ []
0:   retriever_score_scaling ......................... False
0:   retriever_seq_length ............................ 256
0:   reweight_loss_based_on_position_frequency ....... False
0:   sample_rate ..................................... 1.0
0:   save ............................................ checkpoints_14m300b100m
0:   save_interval ................................... 1000
0:   scatter_gather_tensors_in_pipeline .............. True
0:   scattered_embeddings ............................ False
0:   seed ............................................ 1234
0:   seq_length ...................................... 2048
0:   sgd_momentum .................................... 0.9
0:   short_seq_prob .................................. 0.1
0:   skip_train_iteration_range ...................... None
0:   split ........................................... None
0:   split_transformers .............................. False
0:   sync_tp_duplicated_parameters ................... False
0:   synchronize_each_layer .......................... False
0:   tensor_model_parallel_size ...................... 1
0:   tensorboard_dir ................................. tensorboard_14m300b100mval
0:   tensorboard_log_interval ........................ 1
0:   tensorboard_queue_size .......................... 5
0:   test_weighted_split_paths ....................... None
0:   test_weighted_split_paths_path .................. None
0:   tile_factor ..................................... 1
0:   titles_data_path ................................ None
0:   tokenizer_name_or_path .......................... None
0:   tokenizer_type .................................. GPT2BPETokenizer
0:   train_iters ..................................... None
0:   train_samples ................................... 1
0:   train_tokens .................................... None
0:   train_weighted_split_names ...................... ['train']
0:   train_weighted_split_paths ...................... [['/scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_100M_text_document']]
0:   train_weighted_split_paths_path ................. None
0:   train_weighted_split_splits ..................... [['0:1']]
0:   train_weighted_split_weights .................... [['1.0']]
0:   universal_checkpoint ............................ False
0:   use_bnb_optimizer ............................... False
0:   use_checkpoint_lr_scheduler ..................... False
0:   use_contiguous_buffers_in_ddp ................... True
0:   use_cpu_initialization .......................... None
0:   use_one_sent_docs ............................... False
0:   use_pin_memory .................................. False
0:   valid_num_workers ............................... 2
0:   valid_weighted_split_names ...................... ['validation']
0:   valid_weighted_split_paths ...................... [['/scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document']]
0:   valid_weighted_split_paths_path ................. None
0:   valid_weighted_split_splits ..................... [['0:1']]
0:   valid_weighted_split_weights .................... [['1.0']]
0:   virtual_pipeline_model_parallel_size ............ None
0:   vocab_extra_ids ................................. 0
0:   vocab_file ...................................... gpt2/vocab.json
0:   weight_decay .................................... 0.1
0:   world_size ...................................... 8
0:   zero_allgather_bucket_size ...................... 0.0
0:   zero_contigious_gradients ....................... False
0:   zero_reduce_bucket_size ......................... 0.0
0:   zero_reduce_scatter ............................. False
0:   zero_stage ...................................... 0
0: -------------------- end of arguments ---------------------
0: setting number of micro-batches to constant 1
0: > building GPT2BPETokenizer tokenizer ...
0:  > padded vocab (size: 50257) with 47 dummy tokens (new size: 50304)
0: DeepSpeed general environment info:
0: torch install path ............... ['/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch']
0: torch version .................... 1.13.0+rocm5.2
0: torch cuda version ............... None
0: torch hip version ................ 5.2.21151-afdc89f8
0: nvcc version ..................... None
0: deepspeed install path ........... ['/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/deepspeed']
0: deepspeed info ................... 0.7.5, unknown, unknown
0: deepspeed wheel compiled w. ...... torch 1.13, hip 5.1
0: **** Git info for Megatron: git_hash=unknown git_branch=unknown ****
0: > initializing torch distributed ...
0: [2023-04-28 00:43:22,325] [INFO] [comm.py:633:init_distributed] Initializing TorchBackend in DeepSpeed with backend nccl
0: > setting tensorboard ...
0: > initializing tensor model parallel with size 1
0: > initializing pipeline model parallel with size 1
0: > setting random seeds to 1234 ...
0: > initializing model parallel cuda seeds on global rank 0, model parallel rank 0, and data parallel rank 0 with model parallel seed: 3952 and data parallel seed: 1234
0: > compiling dataset index builder ...
0: make: Entering directory '/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/data'
0: make: Nothing to be done for 'default'.
0: make: Leaving directory '/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/data'
0: >>> done with dataset index builder. Compilation time: 0.096 seconds
0: > compiling and loading fused kernels ...
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.cpp -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.cpp [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.h [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_cuda.cu -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.hip [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.h [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.h [skipped, already hipified]
0: Total number of unsupported CUDA function calls: 0
0: 
0: 
0: Total number of replaced kernel launches: 87
0: [1/1] c++ scaled_upper_triang_masked_softmax_hip.cuda.o scaled_upper_triang_masked_softmax_hip.o -shared -L/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/lib -lc10 -lc10_hip -ltorch_cpu -ltorch_hip -ltorch -ltorch_python -L/opt/rocm/lib -lamdhip64 -o scaled_upper_triang_masked_softmax_cuda.so
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax.cpp -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.cpp [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_cuda.cu -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.hip [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.h [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.h [skipped, already hipified]
0: Total number of unsupported CUDA function calls: 0
0: 
0: 
0: Total number of replaced kernel launches: 63
0: [1/1] c++ scaled_masked_softmax_hip.o scaled_masked_softmax_hip.cuda.o -shared -L/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/lib -lc10 -lc10_hip -ltorch_cpu -ltorch_hip -ltorch -ltorch_python -L/opt/rocm/lib -lamdhip64 -o scaled_masked_softmax_cuda.so
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/layer_norm_cuda.cpp -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/layer_norm_cuda.cpp [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/layer_norm_cuda_kernel.cu -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/layer_norm_hip_kernel.hip [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h [skipped, no changes]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.h [skipped, already hipified]
0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.h [skipped, already hipified]
0: Total number of unsupported CUDA function calls: 0
0: 
0: 
0: Total number of replaced kernel launches: 67
0: [1/1] c++ layer_norm_cuda.o layer_norm_hip_kernel.cuda.o -shared -L/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/lib -lc10 -lc10_hip -ltorch_cpu -ltorch_hip -ltorch -ltorch_python -L/opt/rocm/lib -lamdhip64 -o fused_mix_prec_layer_norm_cuda.so
0: >>> done with compiling and loading fused kernels. Compilation time: 11.883 seconds
0: time to initialize megatron (seconds): 6.753
0: [after megatron is initialized] datetime: 2023-04-28 00:43:34 
0: building GPT model ...
0: [2023-04-28 00:43:34,881] [INFO] [utils.py:827:see_memory_usage] Before Building Model
0: [2023-04-28 00:43:34,881] [INFO] [utils.py:828:see_memory_usage] MA 0.0 GB         Max_MA 0.0 GB         CA 0.0 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:34,882] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.7 GB, percent = 7.5%
0: SEED_LAYERS=False BASE_SEED=1234 SEED_FN=None
0: Using topology: {ProcessCoord(pipe=0, data=0, model=0): 0, ProcessCoord(pipe=0, data=1, model=0): 1, ProcessCoord(pipe=0, data=2, model=0): 2, ProcessCoord(pipe=0, data=3, model=0): 3, ProcessCoord(pipe=0, data=4, model=0): 4, ProcessCoord(pipe=0, data=5, model=0): 5, ProcessCoord(pipe=0, data=6, model=0): 6, ProcessCoord(pipe=0, data=7, model=0): 7}
0: [2023-04-28 00:43:35,111] [INFO] [module.py:366:_partition_layers] Partitioning pipeline stages with method type:transformer
0: stage=0 layers=11
0:      0: _to_float16
0:      1: EmbeddingPipe
0:      2: <lambda>
0:      3: ParallelTransformerLayerPipe
0:      4: ParallelTransformerLayerPipe
0:      5: ParallelTransformerLayerPipe
0:      6: ParallelTransformerLayerPipe
0:      7: undo
0:      8: MixedFusedLayerNorm
0:      9: EmbeddingPipe
0:     10: float16_to_fp32
0:   loss: CrossEntropy
0: [2023-04-28 00:43:35,352] [INFO] [utils.py:827:see_memory_usage] After Building Model
0: [2023-04-28 00:43:35,353] [INFO] [utils.py:828:see_memory_usage] MA 0.03 GB         Max_MA 0.03 GB         CA 0.05 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:35,353] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.72 GB, percent = 7.5%
0: setting training iterations to 0
0: > learning rate decay style: cosine
0: DeepSpeed is enabled.
0: [2023-04-28 00:43:35,354] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed info: version=0.7.5, git-hash=unknown, git-branch=unknown
0: [2023-04-28 00:43:40,042] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Flops Profiler Enabled: False
0: [2023-04-28 00:43:40,042] [INFO] [logging.py:68:log_dist] [Rank 0] Removing param_group that has no 'params' in the client Optimizer
0: [2023-04-28 00:43:40,042] [INFO] [logging.py:68:log_dist] [Rank 0] Using client Optimizer as basic optimizer
0: [2023-04-28 00:43:40,043] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Basic Optimizer = FusedAdam
0: [2023-04-28 00:43:40,043] [INFO] [logging.py:68:log_dist] [Rank 0] Creating BF16 optimizer
0: [2023-04-28 00:43:40,161] [INFO] [utils.py:827:see_memory_usage] begin bf16_optimizer
0: [2023-04-28 00:43:40,162] [INFO] [utils.py:828:see_memory_usage] MA 0.03 GB         Max_MA 0.03 GB         CA 0.05 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:40,162] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 40.56 GB, percent = 8.1%
0: ninja: no work to do.
0: Time to load utils op: 0.5600631237030029 secondsTime to load utils op: 0.6787798404693604 seconds
0: 
0: Time to load utils op: 0.6785778999328613 secondsTime to load utils op: 0.6788864135742188 secondsTime to load utils op: 0.6786072254180908 seconds
0: 
0: 
0: Time to load utils op: 0.6782724857330322 secondsTime to load utils op: 0.6787009239196777 seconds
0: 
0: Time to load utils op: 0.6783874034881592 seconds
0: [2023-04-28 00:43:40,836] [INFO] [utils.py:827:see_memory_usage] before initializing group 0
0: [2023-04-28 00:43:40,836] [INFO] [utils.py:828:see_memory_usage] MA 0.03 GB         Max_MA 0.03 GB         CA 0.05 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:40,837] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 39.88 GB, percent = 7.9%
0: [2023-04-28 00:43:41,717] [INFO] [utils.py:827:see_memory_usage] after initializing group 0
0: Time to load utils op: 0.005349636077880859 secondsTime to load utils op: 0.005537986755371094 seconds
0: 
0: Time to load utils op: 0.005392551422119141 secondsTime to load utils op: 0.005532503128051758 seconds
0: 
0: Time to load utils op: 0.005376100540161133 seconds
0: Time to load utils op: 0.005560874938964844 seconds
0: [2023-04-28 00:43:41,717] [INFO] [utils.py:828:see_memory_usage] MA 0.08 GB         Max_MA 0.08 GB         CA 0.12 GB         Max_CA 0 GB 
0: Time to load utils op: 0.0051631927490234375 seconds
0: [2023-04-28 00:43:41,718] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 38.57 GB, percent = 7.7%
0: [2023-04-28 00:43:41,834] [INFO] [utils.py:827:see_memory_usage] before initializing group 1
0: [2023-04-28 00:43:41,834] [INFO] [utils.py:828:see_memory_usage] MA 0.08 GB         Max_MA 0.08 GB         CA 0.12 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:41,835] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 38.51 GB, percent = 7.7%
0: [2023-04-28 00:43:41,940] [INFO] [utils.py:827:see_memory_usage] after initializing group 1
0: [2023-04-28 00:43:41,940] [INFO] [utils.py:828:see_memory_usage] MA 0.09 GB         Max_MA 0.09 GB         CA 0.12 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:41,940] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 38.3 GB, percent = 7.6%
0: [2023-04-28 00:43:42,042] [INFO] [utils.py:827:see_memory_usage] before initializing group 2
0: [2023-04-28 00:43:42,042] [INFO] [utils.py:828:see_memory_usage] MA 0.09 GB         Max_MA 0.09 GB         CA 0.12 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:42,043] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.97 GB, percent = 7.5%
0: [2023-04-28 00:43:42,143] [INFO] [utils.py:827:see_memory_usage] after initializing group 2
0: [2023-04-28 00:43:42,143] [INFO] [utils.py:828:see_memory_usage] MA 0.09 GB         Max_MA 0.09 GB         CA 0.12 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:42,143] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.93 GB, percent = 7.5%
0: [2023-04-28 00:43:42,241] [INFO] [utils.py:827:see_memory_usage] before initialize_optimizer
0: [2023-04-28 00:43:42,242] [INFO] [utils.py:828:see_memory_usage] MA 0.09 GB         Max_MA 0.09 GB         CA 0.12 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:42,242] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.93 GB, percent = 7.5%
0: [2023-04-28 00:43:42,348] [INFO] [utils.py:827:see_memory_usage] end initialize_optimizer
0: [2023-04-28 00:43:42,349] [INFO] [utils.py:828:see_memory_usage] MA 0.1 GB         Max_MA 0.1 GB         CA 0.12 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:42,349] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.93 GB, percent = 7.5%
0: [2023-04-28 00:43:42,449] [INFO] [utils.py:827:see_memory_usage] end bf16_optimizer
0: [2023-04-28 00:43:42,450] [INFO] [utils.py:828:see_memory_usage] MA 0.1 GB         Max_MA 0.1 GB         CA 0.12 GB         Max_CA 0 GB 
0: [2023-04-28 00:43:42,450] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.93 GB, percent = 7.5%
0: [2023-04-28 00:43:42,450] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Final Optimizer = FusedAdam
0: [2023-04-28 00:43:42,450] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed using client LR scheduler
0: [2023-04-28 00:43:42,450] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed LR Scheduler = <megatron.learning_rates.AnnealingLR object at 0x153181a5fdc0>
0: [2023-04-28 00:43:42,450] [INFO] [logging.py:68:log_dist] [Rank 0] step=0, skipped=0, lr=[0.0002, 0.0002, 0.0002], mom=[(0.9, 0.999), (0.9, 0.999), (0.9, 0.999)]
0: [2023-04-28 00:43:42,450] [INFO] [config.py:1007:print] DeepSpeedEngine configuration:
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   activation_checkpointing_config  {
0:     "partition_activations": false, 
0:     "contiguous_memory_optimization": false, 
0:     "cpu_checkpointing": false, 
0:     "number_checkpoints": null, 
0:     "synchronize_checkpoint_boundary": false, 
0:     "profile": false
0: }
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   aio_config ................... {'block_size': 1048576, 'queue_depth': 8, 'thread_count': 1, 'single_submit': False, 'overlap_events': True}
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   amp_enabled .................. False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   amp_params ................... False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   autotuning_config ............ {
0:     "enabled": false, 
0:     "start_step": null, 
0:     "end_step": null, 
0:     "metric_path": null, 
0:     "arg_mappings": null, 
0:     "metric": "throughput", 
0:     "model_info": null, 
0:     "results_dir": "/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/autotuning_results", 
0:     "exps_dir": "/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/autotuning_exps", 
0:     "overwrite": true, 
0:     "fast": true, 
0:     "start_profile_step": 3, 
0:     "end_profile_step": 5, 
0:     "tuner_type": "gridsearch", 
0:     "tuner_early_stopping": 5, 
0:     "tuner_num_trials": 50, 
0:     "model_info_path": null, 
0:     "mp_size": 1, 
0:     "max_train_batch_size": null, 
0:     "min_train_batch_size": 1, 
0:     "max_train_micro_batch_size_per_gpu": 1.024000e+03, 
0:     "min_train_micro_batch_size_per_gpu": 1, 
0:     "num_tuning_micro_batch_sizes": 3
0: }
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   bfloat16_enabled ............. True
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   checkpoint_parallel_write_pipeline  False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   checkpoint_tag_validation_enabled  True
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   checkpoint_tag_validation_fail  False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   comms_config ................. <deepspeed.comm.config.DeepSpeedCommsConfig object at 0x153181a5fb80>
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   communication_data_type ...... None
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   compression_config ........... {'weight_quantization': {'shared_parameters': {'enabled': False, 'quantizer_kernel': False, 'schedule_offset': 0, 'quantize_groups': 1, 'quantize_verbose': False, 'quantization_type': 'symmetric', 'quantize_weight_in_forward': False, 'rounding': 'nearest', 'fp16_mixed_quantize': False, 'quantize_change_ratio': 0.001}, 'different_groups': {}}, 'activation_quantization': {'shared_parameters': {'enabled': False, 'quantization_type': 'symmetric', 'range_calibration': 'dynamic', 'schedule_offset': 1000}, 'different_groups': {}}, 'sparse_pruning': {'shared_parameters': {'enabled': False, 'method': 'l1', 'schedule_offset': 1000}, 'different_groups': {}}, 'row_pruning': {'shared_parameters': {'enabled': False, 'method': 'l1', 'schedule_offset': 1000}, 'different_groups': {}}, 'head_pruning': {'shared_parameters': {'enabled': False, 'method': 'topk', 'schedule_offset': 1000}, 'different_groups': {}}, 'channel_pruning': {'shared_pa
0: rameters': {'enabled': False, 'method': 'l1', 'schedule_offset': 1000}, 'different_groups': {}}, 'layer_reduction': {'enabled': False}}
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   curriculum_enabled ........... False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   curriculum_params ............ False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   dataloader_drop_last ......... False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   disable_allgather ............ False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   dump_state ................... False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   dynamic_loss_scale_args ...... None
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   eigenvalue_enabled ........... False
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   eigenvalue_gas_boundary_resolution  1
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   eigenvalue_layer_name ........ bert.encoder.layer
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   eigenvalue_layer_num ......... 0
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   eigenvalue_max_iter .......... 100
0: [2023-04-28 00:43:42,451] [INFO] [config.py:1011:print]   eigenvalue_stability ......... 1e-06
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   eigenvalue_tol ............... 0.01
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   eigenvalue_verbose ........... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   elasticity_enabled ........... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   flops_profiler_config ........ {
0:     "enabled": false, 
0:     "profile_step": 1, 
0:     "module_depth": -1, 
0:     "top_modules": 1, 
0:     "detailed": true, 
0:     "output_file": null
0: }
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   fp16_auto_cast ............... None
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   fp16_enabled ................. False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   fp16_master_weights_and_gradients  False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   global_rank .................. 0
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   gradient_accumulation_steps .. 1
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   gradient_clipping ............ 1.0
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   gradient_predivide_factor .... 1.0
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   initial_dynamic_scale ........ 1
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   load_universal_checkpoint .... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   loss_scale ................... 1.0
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   memory_breakdown ............. False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   monitor_config ............... <deepspeed.monitor.config.DeepSpeedMonitorConfig object at 0x153181a5faf0>
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   nebula_config ................ {
0:     "enabled": false, 
0:     "persistent_storage_path": null, 
0:     "persistent_time_interval": 100, 
0:     "num_of_version_in_retention": 2, 
0:     "enable_nebula_load": true, 
0:     "load_path": null
0: }
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   optimizer_legacy_fusion ...... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   optimizer_name ............... None
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   optimizer_params ............. None
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   pipeline ..................... {'stages': 'auto', 'partition': 'best', 'seed_layers': False, 'activation_checkpoint_interval': 0}
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   pld_enabled .................. False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   pld_params ................... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   prescale_gradients ........... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   scheduler_name ............... None
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   scheduler_params ............. None
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   sparse_attention ............. None
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   sparse_gradients_enabled ..... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   steps_per_print .............. 2000
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   train_batch_size ............. 256
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   train_micro_batch_size_per_gpu  32
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   use_node_local_storage ....... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   wall_clock_breakdown ......... False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   world_size ................... 8
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   zero_allow_untested_optimizer  False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   zero_config .................. stage=0 contiguous_gradients=True reduce_scatter=True reduce_bucket_size=500000000 allgather_partitions=True allgather_bucket_size=500000000 overlap_comm=False load_from_fp32_weights=True elastic_checkpoint=False offload_param=None offload_optimizer=None sub_group_size=1000000000 cpu_offload_param=None cpu_offload_use_pin_memory=None cpu_offload=None prefetch_bucket_size=50000000 param_persistence_threshold=100000 model_persistence_threshold=9223372036854775807 max_live_parameters=1000000000 max_reuse_distance=1000000000 gather_16bit_weights_on_model_save=False stage3_gather_fp16_weights_on_model_save=False ignore_unused_parameters=True legacy_stage1=False round_robin_gradients=False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   zero_enabled ................. False
0: [2023-04-28 00:43:42,452] [INFO] [config.py:1011:print]   zero_optimization_stage ...... 0
0: [2023-04-28 00:43:42,452] [INFO] [config.py:996:print_user_config]   json = {
0:     "train_micro_batch_size_per_gpu": 32, 
0:     "train_batch_size": 256, 
0:     "gradient_clipping": 1.0, 
0:     "zero_optimization": {
0:         "stage": 0
0:     }, 
0:     "bf16": {
0:         "enabled": true
0:     }, 
0:     "steps_per_print": 2.000000e+03, 
0:     "wall_clock_breakdown": false
0: }
0: Time to load utils op: 0.00037550926208496094 seconds
0: [2023-04-28 00:43:42,453] [INFO] [engine.py:87:__init__] CONFIG: micro_batches=1 micro_batch_size=32
0: [2023-04-28 00:43:42,466] [INFO] [engine.py:145:__init__] RANK=0 STAGE=0 LAYERS=11 [0, 11) STAGE_PARAMS=14147392 (14.147M) TOTAL_PARAMS=14147392 (14.147M) UNIQUE_PARAMS=14147392 (14.147M)
0: [2023-04-28 00:43:42,479] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,479] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,479] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,479] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,479] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,479] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,479] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,479] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,517] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt...
0: [2023-04-28 00:43:42,518] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,518] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,518] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,518] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,518] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/mp_rank_00_model_states.pt.
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,519] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,611] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,611] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,611] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,611] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,611] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,611] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,611] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,611] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,612] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,613] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,613] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,613] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,613] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,613] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,613] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,613] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt...
0: [2023-04-28 00:43:42,625] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,626] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,626] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,626] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,626] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,629] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,631] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,631] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,631] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_01-model_00-model_states.pt.
0: [2023-04-28 00:43:42,631] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,635] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,635] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,635] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,635] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,635] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,636] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,655] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,655] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,656] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt...
0: [2023-04-28 00:43:42,657] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,657] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,657] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,658] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,658] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,658] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,658] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,659] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_03-model_00-model_states.pt.
0: [2023-04-28 00:43:42,659] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,659] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,659] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,659] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,659] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,659] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,660] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,686] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,686] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt...
0: [2023-04-28 00:43:42,687] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,688] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,688] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,688] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,688] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,689] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,689] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,689] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_04-model_00-model_states.pt.
0: [2023-04-28 00:43:42,689] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,690] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,690] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,690] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,690] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,690] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,690] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,713] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,713] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,714] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt...
0: [2023-04-28 00:43:42,715] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,715] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,715] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,715] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,716] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,716] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,716] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,716] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_05-model_00-model_states.pt.
0: [2023-04-28 00:43:42,716] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,717] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,717] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,717] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,717] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,717] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,717] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,741] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,741] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt...
0: [2023-04-28 00:43:42,742] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,743] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,743] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,743] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,744] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,744] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,744] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,744] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_06-model_00-model_states.pt.
0: [2023-04-28 00:43:42,744] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,745] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,745] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,745] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,745] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,745] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,745] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,761] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt...
0: [2023-04-28 00:43:42,762] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,762] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,762] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,762] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,762] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,762] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,762] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0: [2023-04-28 00:43:42,762] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/layer_08-model_00-model_states.pt.
0:  > overriding learning rate value to 0.0002
0:  > overriding minimum learning rate value to 2e-05
0:  > overriding warmup iterations value to 0
0:  > overriding total number of iterations value to 1
0:  > overriding decay style value to cosine
0: [2023-04-28 00:43:42,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt...
0: [2023-04-28 00:43:42,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt...
0: [2023-04-28 00:43:42,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt...
0: [2023-04-28 00:43:42,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt...
0: [2023-04-28 00:43:42,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt...
0: [2023-04-28 00:43:42,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt...
0: [2023-04-28 00:43:42,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt...
0: [2023-04-28 00:43:42,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt...
0: [2023-04-28 00:43:42,837] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt.
0: [2023-04-28 00:43:42,837] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 0
0: [2023-04-28 00:43:42,838] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 0
0: could not find arguments in the checkpoint ...
0:  checkpoint version 3.0
0: [2023-04-28 00:43:42,853] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt.
0: [2023-04-28 00:43:42,853] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 5
0: [2023-04-28 00:43:42,854] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 5
0: [2023-04-28 00:43:42,876] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt.
0: [2023-04-28 00:43:42,876] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 6
0: [2023-04-28 00:43:42,878] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 6
0: [2023-04-28 00:43:42,878] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt.
0: [2023-04-28 00:43:42,878] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt.
0: [2023-04-28 00:43:42,878] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 1
0: [2023-04-28 00:43:42,878] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 4
0: [2023-04-28 00:43:42,879] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt.
0: [2023-04-28 00:43:42,879] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 1
0: [2023-04-28 00:43:42,879] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 3
0: [2023-04-28 00:43:42,880] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 4
0: [2023-04-28 00:43:42,880] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt.
0: [2023-04-28 00:43:42,880] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 7
0: [2023-04-28 00:43:42,881] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 3
0: [2023-04-28 00:43:42,883] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 7
0: [2023-04-28 00:43:42,886] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m300b100m/global_step572204/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt.
0: [2023-04-28 00:43:42,886] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 2
0: [2023-04-28 00:43:42,887] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 2
0:   successfully loaded checkpoint from checkpoints_14m300b100m at iteration 0
0: time (ms) | load-checkpoint: 421.28
0: estimated model parameters: 0.014147392
0: estimated model parameters without embeddings: 0.002420544
0: [after model, optimizer, and learning rate scheduler are built] datetime: 2023-04-28 00:43:43 
0: > building train, validation, and test datasets ...
0:  > datasets target sizes (minimum size):
0:     train:      1
0:     validation: 25600
0:     test:       25600
0: > building train, validation, and test datasets for GPT ...
0:  > building dataset index ...
0:     reading sizes...
0:     reading pointers...
0:     reading document index...
0:     creating numpy buffer of mmap...
0:     creating memory view of numpy buffer...
0:  > finished creating indexed dataset in 0.026684 seconds
0:     number of documents: 208931
0:  > dataset split:
0:     train:
0:      document indices in [0, 208931) total of 208931 documents
0:  > loading doc-idx mapping from /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_100M_text_document_train_indexmap_1ns_2048sl_1234s_doc_idx.npy
0:  > loading sample-idx mapping from /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_100M_text_document_train_indexmap_1ns_2048sl_1234s_sample_idx.npy
0:  > loading shuffle-idx mapping from /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_100M_text_document_train_indexmap_1ns_2048sl_1234s_shuffle_idx.npy
0:     loaded indexed file in 0.101 seconds
0:     total number of samples: 48805
0:     total number of epochs: 1
0:  > building dataset index ...
0:     reading sizes...
0:     reading pointers...
0:     reading document index...
0:     creating numpy buffer of mmap...
0:     creating memory view of numpy buffer...
0:  > finished creating indexed dataset in 0.049837 seconds
0:     number of documents: 364608
0:  > dataset split:
0:     validation:
0:      document indices in [0, 364608) total of 364608 documents
0:  > loading doc-idx mapping from /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document_validation_indexmap_25600ns_2048sl_1234s_doc_idx.npy
0:  > loading sample-idx mapping from /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document_validation_indexmap_25600ns_2048sl_1234s_sample_idx.npy
0:  > loading shuffle-idx mapping from /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document_validation_indexmap_25600ns_2048sl_1234s_shuffle_idx.npy
0:     loaded indexed file in 0.104 seconds
0:     total number of samples: 84978
0:     total number of epochs: 1
0: > finished creating GPT datasets ...
0: time (ms) | model-and-optimizer-setup: 8780.32 | train/valid/test-data-iterators-setup: 4731.61
0: [after dataloaders are built] datetime: 2023-04-28 00:43:48 
0: done with setup ...
0: training ...
0: [after training is done] datetime: 2023-04-28 00:43:48 
0: -----------------------------------------------------------------------------------------------------------------
0: validation loss at the end of training for val data | lm loss value: 4.369110E+00 | lm loss PPL: 7.897332E+01 | 
0: -----------------------------------------------------------------------------------------------------------------
END 3425905: Fri 28 Apr 2023 12:44:25 AM EEST