lm1-misc / 14m200m100m /3423781.out
Muennighoff's picture
A
f4bcf01
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 4 --global-batch-size 64 --train-samples 1 --vocab-file gpt2/vocab.json --merge-file gpt2/merges.txt --clip-grad 1.0 --kill-switch-path kill-switch-14m200m100mval --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_14m200m100mval --tensorboard-queue-size 5 --log-timers-to-tensorboard --log-batch-size-to-tensorboard --log-validation-ppl-to-tensorboard --save checkpoints_14m200m100m --load checkpoints_14m200m100m --train-weighted-split-paths-path train100m.txt --valid-weighted-split-paths-path val.txt --data-impl mmap --deepspeed --deepspeed_config ds_configs/3423781.json --zero-stage 0
START 3423781: Thu 27 Apr 2023 03:53:57 PM EEST
0:
0:
0: ======================= ROCm System Management Interface =======================
0: ================================= Concise Info =================================
0: GPU Temp AvgPwr SCLK MCLK Fan Perf PwrCap VRAM% GPU%
0: 0 43.0c 94.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 40.0c 97.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
0: 3 42.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
0: 4 44.0c 87.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
0: 5 48.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
0: 6 39.0c 90.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 ==============================
1:
1:
1: ======================= ROCm System Management Interface =======================
1: ================================= Concise Info =================================
1: GPU Temp AvgPwr SCLK MCLK Fan Perf PwrCap VRAM% GPU%
1: 0 47.0c 88.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
1: 1 47.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
1: 2 45.0c 91.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
1: 3 48.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
1: 4 45.0c 92.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
1: 5 47.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
1: 6 43.0c 86.0W 800Mhz 1600Mhz 0% auto 560.0W 0% 0%
1: 7 44.0c N/A 800Mhz 1600Mhz 0% auto 0.0W 0% 0%
1: ================================================================================
1: ============================= End of ROCm SMI Log ==============================
0: Launching on nid007280 (0/2), master nid007280 port 9999, GPUs 8, CUDA: True
1: Launching on nid007281 (1/2), master nid007280 port 9999, GPUs 8, CUDA: True
0: using world size: 16, data-parallel-size: 16, 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 .............................. 16
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/3423781.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 ............................... 64
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-14m200m100mval
0: kv_channels ..................................... 32
0: layer_norm_fusion ............................... True
0: layernorm_epsilon ............................... 1e-05
0: lazy_mpu_init ................................... None
0: load ............................................ checkpoints_14m200m100m
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 ................................ 4
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_14m200m100m
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_14m200m100mval
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 ...................................... 16
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
1: > setting tensorboard ...
0: **** Git info for Megatron: git_hash=unknown git_branch=unknown ****
0: > initializing torch distributed ...
0: [2023-04-27 15:56:57,559] [INFO] [comm.py:633:init_distributed] Initializing TorchBackend in DeepSpeed with backend nccl
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.092 seconds
0: > compiling and loading fused kernels ...
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.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_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_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: 102
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: 21.125 seconds
0: time to initialize megatron (seconds): 47.423
0: [after megatron is initialized] datetime: 2023-04-27 15:57:19
0: building GPT model ...
0: [2023-04-27 15:57:19,552] [INFO] [utils.py:827:see_memory_usage] Before Building Model
0: [2023-04-27 15:57:19,552] [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-27 15:57:19,553] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 39.14 GB, percent = 7.8%
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, ProcessCoord(pipe=0, data=8, model=0): 8, ProcessCoord(pipe=0, data=9, model=0): 9, ProcessCoord(pipe=0, data=10, model=0): 10, ProcessCoord(pipe=0, data=11, model=0): 11, ProcessCoord(pipe=0, data=12, model=0): 12, ProcessCoord(pipe=0, data=13, model=0): 13, ProcessCoord(pipe=0, data=14, model=0): 14, ProcessCoord(pipe=0, data=15, model=0): 15}
0: [2023-04-27 15:57:20,022] [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-27 15:57:20,199] [INFO] [utils.py:827:see_memory_usage] After Building Model
0: [2023-04-27 15:57:20,200] [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-27 15:57:20,200] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 39.16 GB, percent = 7.8%
0: setting training iterations to 0
0: > learning rate decay style: cosine
0: DeepSpeed is enabled.
0: [2023-04-27 15:57:20,201] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed info: version=0.7.5, git-hash=unknown, git-branch=unknown
0: [2023-04-27 15:57:28,916] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Flops Profiler Enabled: False
0: [2023-04-27 15:57:28,917] [INFO] [logging.py:68:log_dist] [Rank 0] Removing param_group that has no 'params' in the client Optimizer
0: [2023-04-27 15:57:28,917] [INFO] [logging.py:68:log_dist] [Rank 0] Using client Optimizer as basic optimizer
0: [2023-04-27 15:57:28,917] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Basic Optimizer = FusedAdam
0: [2023-04-27 15:57:28,917] [INFO] [logging.py:68:log_dist] [Rank 0] Creating BF16 optimizer
0: [2023-04-27 15:57:29,047] [INFO] [utils.py:827:see_memory_usage] begin bf16_optimizer
0: [2023-04-27 15:57:29,048] [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-27 15:57:29,048] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 43.34 GB, percent = 8.6%
0: ninja: no work to do.
0: Time to load utils op: 0.4586927890777588 seconds
0: Time to load utils op: 0.4591197967529297 seconds
0: Time to load utils op: 0.4660193920135498 seconds
0: Time to load utils op: 0.45920777320861816 seconds
0: Time to load utils op: 0.46649932861328125 seconds
0: Time to load utils op: 0.4662938117980957 seconds
0: Time to load utils op: 0.45936083793640137 secondsTime to load utils op: 0.35999369621276855 seconds
0:
0: [2023-04-27 15:57:29,526] [INFO] [utils.py:827:see_memory_usage] before initializing group 0
0: [2023-04-27 15:57:29,527] [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-27 15:57:29,527] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 43.02 GB, percent = 8.5%
1: Time to load utils op: 0.6348512172698975 seconds
1: Time to load utils op: 0.6342995166778564 seconds
1: Time to load utils op: 0.6355595588684082 secondsTime to load utils op: 0.6350359916687012 seconds
1:
1: Time to load utils op: 0.6354680061340332 seconds
1: Time to load utils op: 0.6342124938964844 seconds
1: Time to load utils op: 0.634530782699585 secondsTime to load utils op: 0.6346714496612549 seconds
1:
0: Time to load utils op: 0.0028772354125976562 seconds
0: Time to load utils op: 0.006854534149169922 seconds
0: Time to load utils op: 0.006665945053100586 seconds
0: Time to load utils op: 0.006605625152587891 seconds
0: Time to load utils op: 0.007185459136962891 secondsTime to load utils op: 0.007367610931396484 seconds
0:
0: Time to load utils op: 0.007400035858154297 seconds
0: [2023-04-27 15:57:29,775] [INFO] [utils.py:827:see_memory_usage] after initializing group 0
0: [2023-04-27 15:57:29,776] [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-27 15:57:29,776] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 42.81 GB, percent = 8.5%
0: [2023-04-27 15:57:29,884] [INFO] [utils.py:827:see_memory_usage] before initializing group 1
0: [2023-04-27 15:57:29,884] [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-27 15:57:29,885] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 42.92 GB, percent = 8.5%
0: [2023-04-27 15:57:29,990] [INFO] [utils.py:827:see_memory_usage] after initializing group 1
0: [2023-04-27 15:57:29,991] [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-27 15:57:29,991] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 42.76 GB, percent = 8.5%
0: [2023-04-27 15:57:30,095] [INFO] [utils.py:827:see_memory_usage] before initializing group 2
0: [2023-04-27 15:57:30,096] [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-27 15:57:30,096] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 42.63 GB, percent = 8.5%
0: [2023-04-27 15:57:30,201] [INFO] [utils.py:827:see_memory_usage] after initializing group 2
0: [2023-04-27 15:57:30,202] [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-27 15:57:30,202] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 42.52 GB, percent = 8.5%
1: Time to load utils op: 0.0009665489196777344 seconds
1: Time to load utils op: 0.0010173320770263672 seconds
1: Time to load utils op: 0.0012633800506591797 seconds
1: Time to load utils op: 0.0013079643249511719 seconds
1: Time to load utils op: 0.0012621879577636719 seconds
1: Time to load utils op: 0.0012538433074951172 seconds
1: Time to load utils op: 0.0011365413665771484 seconds
1: Time to load utils op: 0.0011448860168457031 seconds
0: [2023-04-27 15:57:30,307] [INFO] [utils.py:827:see_memory_usage] before initialize_optimizer
0: [2023-04-27 15:57:30,307] [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-27 15:57:30,308] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 42.43 GB, percent = 8.4%
0: [2023-04-27 15:57:30,505] [INFO] [utils.py:827:see_memory_usage] end initialize_optimizer
0: [2023-04-27 15:57:30,506] [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-27 15:57:30,506] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 42.18 GB, percent = 8.4%
0: [2023-04-27 15:57:30,610] [INFO] [utils.py:827:see_memory_usage] end bf16_optimizer
0: [2023-04-27 15:57:30,611] [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-27 15:57:30,611] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory: used = 42.04 GB, percent = 8.4%
0: [2023-04-27 15:57:30,611] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Final Optimizer = FusedAdam
0: [2023-04-27 15:57:30,611] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed using client LR scheduler
0: [2023-04-27 15:57:30,611] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed LR Scheduler = <megatron.learning_rates.AnnealingLR object at 0x14d5b31d1fa0>
0: [2023-04-27 15:57:30,611] [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-27 15:57:30,611] [INFO] [config.py:1007:print] DeepSpeedEngine configuration:
0: [2023-04-27 15:57:30,612] [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-27 15:57:30,612] [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-27 15:57:30,612] [INFO] [config.py:1011:print] amp_enabled .................. False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] amp_params ................... False
0: [2023-04-27 15:57:30,612] [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-27 15:57:30,612] [INFO] [config.py:1011:print] bfloat16_enabled ............. True
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] checkpoint_parallel_write_pipeline False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] checkpoint_tag_validation_enabled True
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] checkpoint_tag_validation_fail False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] comms_config ................. <deepspeed.comm.config.DeepSpeedCommsConfig object at 0x14d5b31d1d30>
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] communication_data_type ...... None
0: [2023-04-27 15:57:30,612] [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-27 15:57:30,612] [INFO] [config.py:1011:print] curriculum_enabled ........... False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] curriculum_params ............ False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] dataloader_drop_last ......... False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] disable_allgather ............ False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] dump_state ................... False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] dynamic_loss_scale_args ...... None
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] eigenvalue_enabled ........... False
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] eigenvalue_gas_boundary_resolution 1
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] eigenvalue_layer_name ........ bert.encoder.layer
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] eigenvalue_layer_num ......... 0
0: [2023-04-27 15:57:30,612] [INFO] [config.py:1011:print] eigenvalue_max_iter .......... 100
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] eigenvalue_stability ......... 1e-06
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] eigenvalue_tol ............... 0.01
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] eigenvalue_verbose ........... False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] elasticity_enabled ........... False
0: [2023-04-27 15:57:30,613] [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-27 15:57:30,613] [INFO] [config.py:1011:print] fp16_auto_cast ............... None
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] fp16_enabled ................. False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] fp16_master_weights_and_gradients False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] global_rank .................. 0
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] gradient_accumulation_steps .. 1
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] gradient_clipping ............ 1.0
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] gradient_predivide_factor .... 1.0
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] initial_dynamic_scale ........ 1
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] load_universal_checkpoint .... False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] loss_scale ................... 1.0
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] memory_breakdown ............. False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] monitor_config ............... <deepspeed.monitor.config.DeepSpeedMonitorConfig object at 0x14d5b31d1ca0>
0: [2023-04-27 15:57:30,613] [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-27 15:57:30,613] [INFO] [config.py:1011:print] optimizer_legacy_fusion ...... False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] optimizer_name ............... None
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] optimizer_params ............. None
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] pipeline ..................... {'stages': 'auto', 'partition': 'best', 'seed_layers': False, 'activation_checkpoint_interval': 0}
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] pld_enabled .................. False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] pld_params ................... False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] prescale_gradients ........... False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] scheduler_name ............... None
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] scheduler_params ............. None
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] sparse_attention ............. None
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] sparse_gradients_enabled ..... False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] steps_per_print .............. 2000
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] train_batch_size ............. 64
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] train_micro_batch_size_per_gpu 4
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] use_node_local_storage ....... False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] wall_clock_breakdown ......... False
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] world_size ................... 16
0: [2023-04-27 15:57:30,613] [INFO] [config.py:1011:print] zero_allow_untested_optimizer False
0: [2023-04-27 15:57:30,614] [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-27 15:57:30,614] [INFO] [config.py:1011:print] zero_enabled ................. False
0: [2023-04-27 15:57:30,614] [INFO] [config.py:1011:print] zero_optimization_stage ...... 0
0: [2023-04-27 15:57:30,614] [INFO] [config.py:996:print_user_config] json = {
0: "train_micro_batch_size_per_gpu": 4,
0: "train_batch_size": 64,
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.0004253387451171875 seconds
0: [2023-04-27 15:57:30,614] [INFO] [engine.py:87:__init__] CONFIG: micro_batches=1 micro_batch_size=4
0: [2023-04-27 15:57:30,669] [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)
1: [2023-04-27 15:57:30,674] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,674] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,674] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,674] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,674] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,674] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,674] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,674] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,704] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,704] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,704] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,704] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,704] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,704] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,704] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,705] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
1: [2023-04-27 15:57:30,706] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,706] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,706] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,707] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,708] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,708] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,728] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,728] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,728] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,728] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,728] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,728] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,728] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,728] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,790] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,791] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt...
0: [2023-04-27 15:57:30,792] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,792] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,792] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,792] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,792] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,792] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,792] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
0: [2023-04-27 15:57:30,792] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/mp_rank_00_model_states.pt.
1: [2023-04-27 15:57:30,794] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,794] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,794] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,794] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,794] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,794] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,794] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,794] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,795] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,796] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,796] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,796] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,797] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,797] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,797] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,797] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,797] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,800] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,800] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,800] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,800] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,800] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,800] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,800] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
1: [2023-04-27 15:57:30,802] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,803] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,807] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,807] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,807] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,807] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,807] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,807] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,807] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,808] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,814] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,814] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,814] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,814] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,814] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,814] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,824] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,825] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,825] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,825] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,825] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,826] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,826] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,826] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,826] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
1: [2023-04-27 15:57:30,826] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,826] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,826] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,826] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,827] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,827] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,827] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,827] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,846] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
1: [2023-04-27 15:57:30,847] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,847] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,847] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,847] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,847] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,848] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,848] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,848] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,848] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,848] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,848] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,848] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,849] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,849] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,849] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
1: [2023-04-27 15:57:30,850] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,873] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
1: [2023-04-27 15:57:30,874] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,874] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,874] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,874] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,874] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,874] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,875] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,876] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:30,895] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,895] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,895] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,895] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,895] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,895] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,895] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,897] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,897] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,898] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,898] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,898] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,898] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,898] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,899] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,901] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt...
0: [2023-04-27 15:57:30,910] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,914] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,915] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,916] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
1: [2023-04-27 15:57:30,916] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,916] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,917] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,917] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,917] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,917] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,917] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,917] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
1: [2023-04-27 15:57:30,917] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,917] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,918] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,919] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,920] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,920] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,920] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
1: [2023-04-27 15:57:30,920] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:30,921] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,921] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,921] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,922] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,922] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
1: [2023-04-27 15:57:30,923] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 9
1: [2023-04-27 15:57:30,923] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 10
1: [2023-04-27 15:57:30,923] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 15
1: [2023-04-27 15:57:30,923] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 13
1: [2023-04-27 15:57:30,923] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 14
1: [2023-04-27 15:57:30,923] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 12
1: [2023-04-27 15:57:30,923] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 11
1: [2023-04-27 15:57:30,923] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 8
0: [2023-04-27 15:57:30,924] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_01-model_00-model_states.pt.
0: [2023-04-27 15:57:30,925] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,926] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,926] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,927] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,927] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,928] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,941] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,941] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,941] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,941] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,941] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,942] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,943] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,943] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,943] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt...
0: [2023-04-27 15:57:30,944] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,944] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,944] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,944] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,944] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,944] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,944] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,945] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,946] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,946] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,946] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,946] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,946] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,952] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_03-model_00-model_states.pt.
0: [2023-04-27 15:57:30,954] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,965] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,965] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt...
0: [2023-04-27 15:57:30,966] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,968] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,968] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,968] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,968] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,968] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,968] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,968] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,970] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,970] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,970] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,970] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,970] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,970] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,974] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_04-model_00-model_states.pt.
0: [2023-04-27 15:57:30,975] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,988] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,988] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,989] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt...
0: [2023-04-27 15:57:30,990] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,991] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:30,991] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,991] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,992] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,992] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,992] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,992] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,993] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:30,993] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:30,993] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:30,993] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:30,993] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:30,993] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:30,997] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_05-model_00-model_states.pt.
0: [2023-04-27 15:57:30,998] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,017] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,017] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt...
0: [2023-04-27 15:57:31,018] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,019] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,021] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,021] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,021] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,021] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,021] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,021] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,022] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_06-model_00-model_states.pt.
0: [2023-04-27 15:57:31,022] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,023] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,023] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,023] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,023] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,023] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,023] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,026] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,029] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,029] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt...
0: [2023-04-27 15:57:31,046] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,046] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,046] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,046] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,046] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,046] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/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-27 15:57:31,046] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,046] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/layer_08-model_00-model_states.pt.
0: [2023-04-27 15:57:31,103] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt...
0: [2023-04-27 15:57:31,103] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt...
0: [2023-04-27 15:57:31,103] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt...
0: [2023-04-27 15:57:31,103] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt...
0: [2023-04-27 15:57:31,103] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt...
0: [2023-04-27 15:57:31,103] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt...
0: [2023-04-27 15:57:31,103] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt...
0: [2023-04-27 15:57:31,103] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt...
0: [2023-04-27 15:57:31,208] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt.
0: [2023-04-27 15:57:31,208] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 3
0: [2023-04-27 15:57:31,212] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt.
0: [2023-04-27 15:57:31,212] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 6
0: [2023-04-27 15:57:31,231] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt.
0: [2023-04-27 15:57:31,231] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt.
0: [2023-04-27 15:57:31,231] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt.
0: [2023-04-27 15:57:31,231] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 4
0: [2023-04-27 15:57:31,231] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 1
0: [2023-04-27 15:57:31,231] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 2
0: [2023-04-27 15:57:31,234] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt.
0: [2023-04-27 15:57:31,234] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt.
0: [2023-04-27 15:57:31,235] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 7
0: [2023-04-27 15:57:31,235] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 5
0: [2023-04-27 15:57:31,450] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_14m200m100m/global_step381/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt.
0: [2023-04-27 15:57:31,450] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 0