lm1-misc / 14m600b100m /3448563.out
Muennighoff's picture
Add
20a9f7d
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-14m600b100mval --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_14m600b100mval --tensorboard-queue-size 5 --log-timers-to-tensorboard --log-batch-size-to-tensorboard --log-validation-ppl-to-tensorboard --save checkpoints_14m600b100m --load checkpoints_14m600b100m --train-weighted-split-paths-path train100m.txt --valid-weighted-split-paths-path val.txt --data-impl mmap --deepspeed --deepspeed_config ds_configs/3448563.json --zero-stage 0
START 3448563: Wed 03 May 2023 08:34:26 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 50.0c 88.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
0: 1 45.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
0: 2 41.0c 83.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
0: 3 45.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
0: 4 42.0c 84.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
0: 5 45.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
0: 6 39.0c 82.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
0: 7 41.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
0: ================================================================================
0: ============================= End of ROCm SMI Log ==============================
0: Launching on nid007329 (0/1), master nid007329 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/3448563.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-14m600b100mval
0: kv_channels ..................................... 32
0: layer_norm_fusion ............................... True
0: layernorm_epsilon ............................... 1e-05
0: lazy_mpu_init ................................... None
0: load ............................................ checkpoints_14m600b100m
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_14m600b100m
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_14m600b100mval
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-05-03 08:36:37,822] [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.085 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: ninja: no work to do.
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: ninja: no work to do.
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: 10.642 seconds
0: time to initialize megatron (seconds): -14.877
0: [after megatron is initialized] datetime: 2023-05-03 08:36:49
0: building GPT model ...
0: [2023-05-03 08:36:49,249] [INFO] [utils.py:827:see_memory_usage] Before Building Model
0: [2023-05-03 08:36:49,250] [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-05-03 08:36:49,250] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 38.2 GB, percent = 7.6%
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-05-03 08:36:49,480] [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-05-03 08:36:49,750] [INFO] [utils.py:827:see_memory_usage] After Building Model
0: [2023-05-03 08:36:49,750] [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-05-03 08:36:49,751] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 38.21 GB, percent = 7.6%
0: setting training iterations to 0
0: > learning rate decay style: cosine
0: DeepSpeed is enabled.
0: [2023-05-03 08:36:49,751] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed info: version=0.7.5, git-hash=unknown, git-branch=unknown
0: [2023-05-03 08:36:54,658] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Flops Profiler Enabled: False
0: [2023-05-03 08:36:54,658] [INFO] [logging.py:68:log_dist] [Rank 0] Removing param_group that has no 'params' in the client Optimizer
0: [2023-05-03 08:36:54,658] [INFO] [logging.py:68:log_dist] [Rank 0] Using client Optimizer as basic optimizer
0: [2023-05-03 08:36:54,659] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Basic Optimizer = FusedAdam
0: [2023-05-03 08:36:54,659] [INFO] [logging.py:68:log_dist] [Rank 0] Creating BF16 optimizer
0: [2023-05-03 08:36:54,773] [INFO] [utils.py:827:see_memory_usage] begin bf16_optimizer
0: [2023-05-03 08:36:54,773] [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-05-03 08:36:54,773] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 41.37 GB, percent = 8.2%
0: ninja: no work to do.
0: Time to load utils op: 0.5206212997436523 seconds
0: Time to load utils op: 0.6343584060668945 seconds
0: Time to load utils op: 0.6349160671234131 secondsTime to load utils op: 0.635321855545044 seconds
0:
0: Time to load utils op: 0.634469747543335 seconds
0: Time to load utils op: 0.6343989372253418 secondsTime to load utils op: 0.6343097686767578 seconds
0:
0: Time to load utils op: 0.6350526809692383 seconds
0: [2023-05-03 08:36:55,407] [INFO] [utils.py:827:see_memory_usage] before initializing group 0
0: [2023-05-03 08:36:55,408] [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-05-03 08:36:55,408] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 40.7 GB, percent = 8.1%
0: Time to load utils op: 0.0027306079864501953 seconds
0: Time to load utils op: 0.002711057662963867 seconds
0: Time to load utils op: 0.0027027130126953125 seconds
0: Time to load utils op: 0.0025589466094970703 seconds
0: Time to load utils op: 0.0027480125427246094 seconds
0: Time to load utils op: 0.002451181411743164 secondsTime to load utils op: 0.0027425289154052734 seconds
0:
0: [2023-05-03 08:36:56,184] [INFO] [utils.py:827:see_memory_usage] after initializing group 0
0: [2023-05-03 08:36:56,185] [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-05-03 08:36:56,185] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 39.65 GB, percent = 7.9%
0: [2023-05-03 08:36:56,294] [INFO] [utils.py:827:see_memory_usage] before initializing group 1
0: [2023-05-03 08:36:56,294] [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-05-03 08:36:56,294] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 39.61 GB, percent = 7.9%
0: [2023-05-03 08:36:56,400] [INFO] [utils.py:827:see_memory_usage] after initializing group 1
0: [2023-05-03 08:36:56,400] [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-05-03 08:36:56,401] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 39.44 GB, percent = 7.8%
0: [2023-05-03 08:36:56,503] [INFO] [utils.py:827:see_memory_usage] before initializing group 2
0: [2023-05-03 08:36:56,503] [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-05-03 08:36:56,503] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 39.24 GB, percent = 7.8%
0: [2023-05-03 08:36:56,606] [INFO] [utils.py:827:see_memory_usage] after initializing group 2
0: [2023-05-03 08:36:56,607] [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-05-03 08:36:56,607] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 39.05 GB, percent = 7.8%
0: [2023-05-03 08:36:56,708] [INFO] [utils.py:827:see_memory_usage] before initialize_optimizer
0: [2023-05-03 08:36:56,709] [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-05-03 08:36:56,709] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 38.9 GB, percent = 7.7%
0: [2023-05-03 08:36:56,861] [INFO] [utils.py:827:see_memory_usage] end initialize_optimizer
0: [2023-05-03 08:36:56,862] [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-05-03 08:36:56,862] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 38.67 GB, percent = 7.7%
0: [2023-05-03 08:36:56,964] [INFO] [utils.py:827:see_memory_usage] end bf16_optimizer
0: [2023-05-03 08:36:56,964] [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-05-03 08:36:56,965] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 38.53 GB, percent = 7.7%
0: [2023-05-03 08:36:56,965] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Final Optimizer = FusedAdam
0: [2023-05-03 08:36:56,965] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed using client LR scheduler
0: [2023-05-03 08:36:56,965] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed LR Scheduler = <megatron.learning_rates.AnnealingLR object at 0x154ae674ffd0>
0: [2023-05-03 08:36:56,965] [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-05-03 08:36:56,965] [INFO] [config.py:1007:print] DeepSpeedEngine configuration:
0: [2023-05-03 08:36:56,965] [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-05-03 08:36:56,966] [INFO] [config.py:1011:print] aio_config ................... {'block_size': 1048576, 'queue_depth': 8, 'thread_count': 1, 'single_submit': False, 'overlap_events': True}
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] amp_enabled .................. False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] amp_params ................... False
0: [2023-05-03 08:36:56,966] [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-05-03 08:36:56,966] [INFO] [config.py:1011:print] bfloat16_enabled ............. True
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] checkpoint_parallel_write_pipeline False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] checkpoint_tag_validation_enabled True
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] checkpoint_tag_validation_fail False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] comms_config ................. <deepspeed.comm.config.DeepSpeedCommsConfig object at 0x154ae674fd30>
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] communication_data_type ...... None
0: [2023-05-03 08:36:56,966] [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-05-03 08:36:56,966] [INFO] [config.py:1011:print] curriculum_enabled ........... False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] curriculum_params ............ False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] dataloader_drop_last ......... False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] disable_allgather ............ False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] dump_state ................... False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] dynamic_loss_scale_args ...... None
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] eigenvalue_enabled ........... False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] eigenvalue_gas_boundary_resolution 1
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] eigenvalue_layer_name ........ bert.encoder.layer
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] eigenvalue_layer_num ......... 0
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] eigenvalue_max_iter .......... 100
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] eigenvalue_stability ......... 1e-06
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] eigenvalue_tol ............... 0.01
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] eigenvalue_verbose ........... False
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] elasticity_enabled ........... False
0: [2023-05-03 08:36:56,966] [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-05-03 08:36:56,966] [INFO] [config.py:1011:print] fp16_auto_cast ............... None
0: [2023-05-03 08:36:56,966] [INFO] [config.py:1011:print] fp16_enabled ................. False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] fp16_master_weights_and_gradients False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] global_rank .................. 0
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] gradient_accumulation_steps .. 1
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] gradient_clipping ............ 1.0
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] gradient_predivide_factor .... 1.0
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] initial_dynamic_scale ........ 1
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] load_universal_checkpoint .... False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] loss_scale ................... 1.0
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] memory_breakdown ............. False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] monitor_config ............... <deepspeed.monitor.config.DeepSpeedMonitorConfig object at 0x154ae674fca0>
0: [2023-05-03 08:36:56,967] [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-05-03 08:36:56,967] [INFO] [config.py:1011:print] optimizer_legacy_fusion ...... False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] optimizer_name ............... None
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] optimizer_params ............. None
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] pipeline ..................... {'stages': 'auto', 'partition': 'best', 'seed_layers': False, 'activation_checkpoint_interval': 0}
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] pld_enabled .................. False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] pld_params ................... False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] prescale_gradients ........... False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] scheduler_name ............... None
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] scheduler_params ............. None
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] sparse_attention ............. None
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] sparse_gradients_enabled ..... False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] steps_per_print .............. 2000
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] train_batch_size ............. 256
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] train_micro_batch_size_per_gpu 32
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] use_node_local_storage ....... False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] wall_clock_breakdown ......... False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] world_size ................... 8
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] zero_allow_untested_optimizer False
0: [2023-05-03 08:36:56,967] [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-05-03 08:36:56,967] [INFO] [config.py:1011:print] zero_enabled ................. False
0: [2023-05-03 08:36:56,967] [INFO] [config.py:1011:print] zero_optimization_stage ...... 0
0: [2023-05-03 08:36:56,967] [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.00041031837463378906 seconds
0: [2023-05-03 08:36:56,968] [INFO] [engine.py:87:__init__] CONFIG: micro_batches=1 micro_batch_size=32
0: [2023-05-03 08:36:57,061] [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-05-03 08:36:57,101] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,101] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,101] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,101] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,101] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,101] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,101] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,101] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,141] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt...
0: [2023-05-03 08:36:57,142] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,142] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,142] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,142] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,142] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,142] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/mp_rank_00_model_states.pt.
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,143] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,266] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,266] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,266] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,266] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,266] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,266] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,266] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,267] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,268] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,268] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,268] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,268] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,268] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,268] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,268] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,268] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt...
0: [2023-05-03 08:36:57,283] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,283] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,283] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,283] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,284] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,284] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,284] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,284] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_01-model_00-model_states.pt.
0: [2023-05-03 08:36:57,287] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,289] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,290] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,290] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,290] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,290] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,290] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,290] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,339] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt...
0: [2023-05-03 08:36:57,340] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,340] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,340] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,340] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,340] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,340] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,340] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,341] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_03-model_00-model_states.pt.
0: [2023-05-03 08:36:57,342] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,342] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,342] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,342] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,342] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,342] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,342] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,342] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,370] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt...
0: [2023-05-03 08:36:57,371] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,371] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,371] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,371] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,371] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,371] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,371] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,371] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_04-model_00-model_states.pt.
0: [2023-05-03 08:36:57,372] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,373] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,373] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,373] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,373] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,373] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,373] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,373] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,419] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,420] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,420] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt...
0: [2023-05-03 08:36:57,421] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,421] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,421] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,421] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,421] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,421] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,421] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,422] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_05-model_00-model_states.pt.
0: [2023-05-03 08:36:57,422] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,422] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,422] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,422] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,423] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,423] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,423] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,423] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,442] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,442] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,442] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,442] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,442] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,442] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,442] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,443] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt...
0: [2023-05-03 08:36:57,444] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,444] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,444] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,445] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,445] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,445] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,445] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,445] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,445] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_06-model_00-model_states.pt.
0: [2023-05-03 08:36:57,446] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,446] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,446] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,446] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,446] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,446] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,447] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,459] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,460] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,460] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt...
0: [2023-05-03 08:36:57,460] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,460] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,460] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,460] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/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-05-03 08:36:57,460] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,461] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,461] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,461] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/layer_08-model_00-model_states.pt.
0: [2023-05-03 08:36:57,464] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt...
0: [2023-05-03 08:36:57,464] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt...
0: [2023-05-03 08:36:57,464] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt...
0: [2023-05-03 08:36:57,464] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt...
0: [2023-05-03 08:36:57,464] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt...
0: [2023-05-03 08:36:57,464] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt...
0: [2023-05-03 08:36:57,464] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt...
0: [2023-05-03 08:36:57,464] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt...
0: [2023-05-03 08:36:57,565] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt.
0: [2023-05-03 08:36:57,565] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 5
0: [2023-05-03 08:36:57,566] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 5
0: [2023-05-03 08:36:57,567] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt.
0: [2023-05-03 08:36:57,568] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 4
0: [2023-05-03 08:36:57,568] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt.
0: [2023-05-03 08:36:57,569] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 3
0: [2023-05-03 08:36:57,569] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 4
0: [2023-05-03 08:36:57,570] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt.
0: [2023-05-03 08:36:57,570] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 1
0: [2023-05-03 08:36:57,571] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 3
0: [2023-05-03 08:36:57,572] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 1
0: [2023-05-03 08:36:57,574] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt.
0: [2023-05-03 08:36:57,574] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt.
0: [2023-05-03 08:36:57,574] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 0
0: [2023-05-03 08:36:57,574] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 6
0: [2023-05-03 08:36:57,575] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 0
0: [2023-05-03 08:36:57,575] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 6
0: could not find arguments in the checkpoint ...
0: checkpoint version 3.0
0: [2023-05-03 08:36:57,589] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt.
0: [2023-05-03 08:36:57,589] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 2
0: [2023-05-03 08:36:57,589] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m600b100m/global_step1144409/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt.
0: [2023-05-03 08:36:57,589] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 7
0: [2023-05-03 08:36:57,590] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 2
0: [2023-05-03 08:36:57,590] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 7
0: successfully loaded checkpoint from checkpoints_14m600b100m at iteration 0
0: time (ms) | load-checkpoint: 535.25
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-05-03 08:36:58
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.029365 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.093 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.104306 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.115 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: 9157.75 | train/valid/test-data-iterators-setup: 7071.28
0: [after dataloaders are built] datetime: 2023-05-03 08:37:05
0: done with setup ...
0: training ...
0: [after training is done] datetime: 2023-05-03 08:37:05
0: -----------------------------------------------------------------------------------------------------------------
0: validation loss at the end of training for val data | lm loss value: 4.331469E+00 | lm loss PPL: 7.605590E+01 |
0: -----------------------------------------------------------------------------------------------------------------
END 3448563: Wed 03 May 2023 08:37:40 AM EEST