auto-sync epoch_0_step_3499 2026-09-08 09:59
Browse files- .gitattributes +10 -0
- ckpt/epoch_0_step_3499/config.yaml +88 -0
- ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_0.pt +3 -0
- ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_1.pt +3 -0
- ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_2.pt +3 -0
- ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_3.pt +3 -0
- ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_4.pt +3 -0
- ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_5.pt +3 -0
- ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_6.pt +3 -0
- ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_7.pt +3 -0
- ckpt/epoch_0_step_3499/losses.json +1 -0
- ckpt/epoch_0_step_3499/model/.hf_metadata/config.json +218 -0
- ckpt/epoch_0_step_3499/model/.hf_metadata/fqn_to_dtype_mapping.json +0 -0
- ckpt/epoch_0_step_3499/model/.hf_metadata/fqn_to_file_index_mapping.json +0 -0
- ckpt/epoch_0_step_3499/model/.hf_metadata/generation_config.json +10 -0
- ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json +3 -0
- ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json.ok +0 -0
- ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer_config.json +34 -0
- ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer_config.json.ok +0 -0
- ckpt/epoch_0_step_3499/model/consolidate.sh +46 -0
- ckpt/epoch_0_step_3499/model/shard-00001-model-00001-of-00001.safetensors +3 -0
- ckpt/epoch_0_step_3499/model/shard-00002-model-00001-of-00001.safetensors +3 -0
- ckpt/epoch_0_step_3499/model/shard-00003-model-00001-of-00001.safetensors +3 -0
- ckpt/epoch_0_step_3499/model/shard-00004-model-00001-of-00001.safetensors +3 -0
- ckpt/epoch_0_step_3499/model/shard-00005-model-00001-of-00001.safetensors +3 -0
- ckpt/epoch_0_step_3499/model/shard-00006-model-00001-of-00001.safetensors +3 -0
- ckpt/epoch_0_step_3499/model/shard-00007-model-00001-of-00001.safetensors +3 -0
- ckpt/epoch_0_step_3499/model/shard-00008-model-00001-of-00001.safetensors +3 -0
- ckpt/epoch_0_step_3499/optim/.metadata +3 -0
- ckpt/epoch_0_step_3499/optim/__0_0.distcp +3 -0
- ckpt/epoch_0_step_3499/optim/__1_0.distcp +3 -0
- ckpt/epoch_0_step_3499/optim/__2_0.distcp +3 -0
- ckpt/epoch_0_step_3499/optim/__3_0.distcp +3 -0
- ckpt/epoch_0_step_3499/optim/__4_0.distcp +3 -0
- ckpt/epoch_0_step_3499/optim/__5_0.distcp +3 -0
- ckpt/epoch_0_step_3499/optim/__6_0.distcp +3 -0
- ckpt/epoch_0_step_3499/optim/__7_0.distcp +3 -0
- ckpt/epoch_0_step_3499/rng/rng_global_rank_0.pt +3 -0
- ckpt/epoch_0_step_3499/rng/rng_global_rank_1.pt +3 -0
- ckpt/epoch_0_step_3499/rng/rng_global_rank_2.pt +3 -0
- ckpt/epoch_0_step_3499/rng/rng_global_rank_3.pt +3 -0
- ckpt/epoch_0_step_3499/rng/rng_global_rank_4.pt +3 -0
- ckpt/epoch_0_step_3499/rng/rng_global_rank_5.pt +3 -0
- ckpt/epoch_0_step_3499/rng/rng_global_rank_6.pt +3 -0
- ckpt/epoch_0_step_3499/rng/rng_global_rank_7.pt +3 -0
- ckpt/epoch_0_step_3499/step_scheduler.pt +3 -0
.gitattributes
CHANGED
|
@@ -95,3 +95,13 @@ ckpt/epoch_0_step_2999/optim/__4_0.distcp filter=lfs diff=lfs merge=lfs -text
|
|
| 95 |
ckpt/epoch_0_step_2999/optim/__5_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 96 |
ckpt/epoch_0_step_2999/optim/__6_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 97 |
ckpt/epoch_0_step_2999/optim/__7_0.distcp filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 95 |
ckpt/epoch_0_step_2999/optim/__5_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 96 |
ckpt/epoch_0_step_2999/optim/__6_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 97 |
ckpt/epoch_0_step_2999/optim/__7_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 98 |
+
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
| 99 |
+
ckpt/epoch_0_step_3499/optim/.metadata filter=lfs diff=lfs merge=lfs -text
|
| 100 |
+
ckpt/epoch_0_step_3499/optim/__0_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 101 |
+
ckpt/epoch_0_step_3499/optim/__1_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 102 |
+
ckpt/epoch_0_step_3499/optim/__2_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 103 |
+
ckpt/epoch_0_step_3499/optim/__3_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 104 |
+
ckpt/epoch_0_step_3499/optim/__4_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 105 |
+
ckpt/epoch_0_step_3499/optim/__5_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 106 |
+
ckpt/epoch_0_step_3499/optim/__6_0.distcp filter=lfs diff=lfs merge=lfs -text
|
| 107 |
+
ckpt/epoch_0_step_3499/optim/__7_0.distcp filter=lfs diff=lfs merge=lfs -text
|
ckpt/epoch_0_step_3499/config.yaml
ADDED
|
@@ -0,0 +1,88 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
recipe: TrainFinetuneRecipeForNextTokenPrediction
|
| 2 |
+
seed: 1234
|
| 3 |
+
step_scheduler:
|
| 4 |
+
global_batch_size: 64
|
| 5 |
+
local_batch_size: 1
|
| 6 |
+
ckpt_every_steps: 500
|
| 7 |
+
gc_every_steps: 10
|
| 8 |
+
num_epochs: 1
|
| 9 |
+
max_steps: 4000
|
| 10 |
+
dist_env:
|
| 11 |
+
backend: nccl
|
| 12 |
+
timeout_minutes: 30
|
| 13 |
+
rng:
|
| 14 |
+
_target_: nemo_automodel.components.training.rng.StatefulRNG
|
| 15 |
+
seed: 1234
|
| 16 |
+
ranked: true
|
| 17 |
+
model:
|
| 18 |
+
_target_: nemo_automodel.NeMoAutoModelForCausalLM.from_pretrained
|
| 19 |
+
pretrained_model_name_or_path: /llm-align/open_models/gemma4/gemma-4-E2B
|
| 20 |
+
torch_dtype: torch.bfloat16
|
| 21 |
+
use_liger_kernel: false
|
| 22 |
+
use_sdpa_patching: false
|
| 23 |
+
attn_implementation: sdpa
|
| 24 |
+
text_config:
|
| 25 |
+
use_cache: false
|
| 26 |
+
output_hidden_states: true
|
| 27 |
+
checkpoint:
|
| 28 |
+
enabled: true
|
| 29 |
+
checkpoint_dir: /llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt
|
| 30 |
+
model_save_format: safetensors
|
| 31 |
+
save_consolidated: false
|
| 32 |
+
distributed:
|
| 33 |
+
strategy: fsdp2
|
| 34 |
+
dp_size: none
|
| 35 |
+
tp_size: 1
|
| 36 |
+
cp_size: 1
|
| 37 |
+
pp_size: 1
|
| 38 |
+
sequence_parallel: false
|
| 39 |
+
activation_checkpointing: false
|
| 40 |
+
loss_fn:
|
| 41 |
+
_target_: nemo_automodel.components.loss.linear_ce.FusedLinearCrossEntropy
|
| 42 |
+
logit_softcapping: 30.0
|
| 43 |
+
dataset:
|
| 44 |
+
_target_: nemo_automodel.components.datasets.llm.nanogpt_dataset.NanogptDataset
|
| 45 |
+
file_pattern: /llm-align/liuchonghan/datasets/fineweb_edu_gemma4/train_max_tokens_1200000000/dataset_*.bin
|
| 46 |
+
seq_len: 4096
|
| 47 |
+
shuffle_files: false
|
| 48 |
+
align_to_bos: false
|
| 49 |
+
dataloader:
|
| 50 |
+
_target_: torchdata.stateful_dataloader.StatefulDataLoader
|
| 51 |
+
shuffle: false
|
| 52 |
+
collate_fn: nemo_automodel.components.datasets.utils.default_collater
|
| 53 |
+
num_workers: 2
|
| 54 |
+
optimizer:
|
| 55 |
+
_target_: transformer_engine.pytorch.optimizers.fused_adam.FusedAdam
|
| 56 |
+
betas:
|
| 57 |
+
- 0.9
|
| 58 |
+
- 0.95
|
| 59 |
+
eps: 1.0e-08
|
| 60 |
+
lr: 1.0e-05
|
| 61 |
+
weight_decay: 0.1
|
| 62 |
+
adam_w_mode: true
|
| 63 |
+
bias_correction: true
|
| 64 |
+
master_weights: true
|
| 65 |
+
master_weight_dtype: torch.float32
|
| 66 |
+
store_param_remainders: true
|
| 67 |
+
exp_avg_dtype: torch.float32
|
| 68 |
+
exp_avg_sq_dtype: torch.float32
|
| 69 |
+
lr_scheduler:
|
| 70 |
+
lr_decay_style: cosine
|
| 71 |
+
lr_warmup_steps: 400
|
| 72 |
+
min_lr: 1.0e-06
|
| 73 |
+
clip_grad_norm:
|
| 74 |
+
max_norm: 1.0
|
| 75 |
+
wandb:
|
| 76 |
+
enable: true
|
| 77 |
+
mode: offline
|
| 78 |
+
dir: /llm-align/liuchonghan/runs/gemma4_ngram/control/wandb
|
| 79 |
+
project: gemma4_e2b_ngram
|
| 80 |
+
name: e2b_base_control_fineweb_1.05Btok
|
| 81 |
+
entity: khazzz1c
|
| 82 |
+
freeze_config:
|
| 83 |
+
freeze_modules:
|
| 84 |
+
- path: model.language_model.embed_tokens_per_layer
|
| 85 |
+
- path: model.language_model.embed_tokens
|
| 86 |
+
freeze_vision_tower: true
|
| 87 |
+
freeze_audio_tower: true
|
| 88 |
+
freeze_language_model: false
|
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bb793f081489f96f580463924fac4426ee07bb598caa12721d571c7c1173101b
|
| 3 |
+
size 1915
|
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_1.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:05c348848e2ad250e68b8cf3608afb4930c88aafae13a351cee8fe574604f9aa
|
| 3 |
+
size 1915
|
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_2.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f0858fbac2268b25265f98ecfed2fe0f961516276a9d32270c276a2f67d7d2d
|
| 3 |
+
size 1915
|
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_3.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:63cb0fd67044e2c7469a5ec3be9bd07ac59fc6f522f2e7c4a1720d6b8cc08499
|
| 3 |
+
size 1915
|
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_4.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b95ed75e24ed0200f8e6adcfb3927f1954957222169d0827a49241c1cd4ad9c1
|
| 3 |
+
size 1915
|
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_5.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b68ba1cce41af6a838b98c2ec5e892d06e52af31d2a198e7f396968d4b490e05
|
| 3 |
+
size 1915
|
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_6.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8c084390749fe61d3defff44697d3acbc21a4acfd3fee7fb5f8ac0c066c7f421
|
| 3 |
+
size 1915
|
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_7.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d8ca9245d1631d534e14d6952169389a88d4296083b7d919465ab26a2328b1cc
|
| 3 |
+
size 1915
|
ckpt/epoch_0_step_3499/losses.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"train_loss": 2.3248672485351562}
|
ckpt/epoch_0_step_3499/model/.hf_metadata/config.json
ADDED
|
@@ -0,0 +1,218 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "/llm-align/open_models/gemma4/gemma-4-E2B",
|
| 3 |
+
"architectures": [
|
| 4 |
+
"Gemma4ForConditionalGeneration"
|
| 5 |
+
],
|
| 6 |
+
"audio_config": {
|
| 7 |
+
"_name_or_path": "",
|
| 8 |
+
"architectures": null,
|
| 9 |
+
"attention_chunk_size": 12,
|
| 10 |
+
"attention_context_left": 13,
|
| 11 |
+
"attention_context_right": 0,
|
| 12 |
+
"attention_invalid_logits_value": -1000000000.0,
|
| 13 |
+
"attention_logit_cap": 50.0,
|
| 14 |
+
"chunk_size_feed_forward": 0,
|
| 15 |
+
"conv_kernel_size": 5,
|
| 16 |
+
"dtype": "bfloat16",
|
| 17 |
+
"gradient_clipping": 10000000000.0,
|
| 18 |
+
"hidden_act": "silu",
|
| 19 |
+
"hidden_size": 1024,
|
| 20 |
+
"id2label": {
|
| 21 |
+
"0": "LABEL_0",
|
| 22 |
+
"1": "LABEL_1"
|
| 23 |
+
},
|
| 24 |
+
"initializer_range": 0.02,
|
| 25 |
+
"is_encoder_decoder": false,
|
| 26 |
+
"label2id": {
|
| 27 |
+
"LABEL_0": 0,
|
| 28 |
+
"LABEL_1": 1
|
| 29 |
+
},
|
| 30 |
+
"model_type": "gemma4_audio",
|
| 31 |
+
"num_attention_heads": 8,
|
| 32 |
+
"num_hidden_layers": 12,
|
| 33 |
+
"output_attentions": false,
|
| 34 |
+
"output_hidden_states": false,
|
| 35 |
+
"output_proj_dims": 1536,
|
| 36 |
+
"problem_type": null,
|
| 37 |
+
"residual_weight": 0.5,
|
| 38 |
+
"return_dict": true,
|
| 39 |
+
"rms_norm_eps": 1e-06,
|
| 40 |
+
"subsampling_conv_channels": [
|
| 41 |
+
128,
|
| 42 |
+
32
|
| 43 |
+
],
|
| 44 |
+
"use_clipped_linears": true
|
| 45 |
+
},
|
| 46 |
+
"audio_token_id": 258881,
|
| 47 |
+
"boa_token_id": 256000,
|
| 48 |
+
"boi_token_id": 255999,
|
| 49 |
+
"chunk_size_feed_forward": 0,
|
| 50 |
+
"dtype": "bfloat16",
|
| 51 |
+
"eoa_token_id": 258883,
|
| 52 |
+
"eoa_token_index": 258883,
|
| 53 |
+
"eoi_token_id": 258882,
|
| 54 |
+
"id2label": {
|
| 55 |
+
"0": "LABEL_0",
|
| 56 |
+
"1": "LABEL_1"
|
| 57 |
+
},
|
| 58 |
+
"image_token_id": 258880,
|
| 59 |
+
"initializer_range": 0.02,
|
| 60 |
+
"is_encoder_decoder": false,
|
| 61 |
+
"label2id": {
|
| 62 |
+
"LABEL_0": 0,
|
| 63 |
+
"LABEL_1": 1
|
| 64 |
+
},
|
| 65 |
+
"model_type": "gemma4",
|
| 66 |
+
"output_attentions": false,
|
| 67 |
+
"output_hidden_states": false,
|
| 68 |
+
"problem_type": null,
|
| 69 |
+
"return_dict": true,
|
| 70 |
+
"text_config": {
|
| 71 |
+
"_name_or_path": "",
|
| 72 |
+
"architectures": null,
|
| 73 |
+
"attention_bias": false,
|
| 74 |
+
"attention_dropout": 0.0,
|
| 75 |
+
"attention_k_eq_v": false,
|
| 76 |
+
"bos_token_id": 2,
|
| 77 |
+
"chunk_size_feed_forward": 0,
|
| 78 |
+
"dtype": "bfloat16",
|
| 79 |
+
"enable_moe_block": false,
|
| 80 |
+
"eos_token_id": 1,
|
| 81 |
+
"expert_intermediate_size": null,
|
| 82 |
+
"final_logit_softcapping": 30.0,
|
| 83 |
+
"global_head_dim": 512,
|
| 84 |
+
"head_dim": 256,
|
| 85 |
+
"hidden_activation": "gelu_pytorch_tanh",
|
| 86 |
+
"hidden_size": 1536,
|
| 87 |
+
"hidden_size_per_layer_input": 256,
|
| 88 |
+
"id2label": {
|
| 89 |
+
"0": "LABEL_0",
|
| 90 |
+
"1": "LABEL_1"
|
| 91 |
+
},
|
| 92 |
+
"initializer_range": 0.02,
|
| 93 |
+
"intermediate_size": 6144,
|
| 94 |
+
"is_encoder_decoder": false,
|
| 95 |
+
"label2id": {
|
| 96 |
+
"LABEL_0": 0,
|
| 97 |
+
"LABEL_1": 1
|
| 98 |
+
},
|
| 99 |
+
"layer_types": [
|
| 100 |
+
"sliding_attention",
|
| 101 |
+
"sliding_attention",
|
| 102 |
+
"sliding_attention",
|
| 103 |
+
"sliding_attention",
|
| 104 |
+
"full_attention",
|
| 105 |
+
"sliding_attention",
|
| 106 |
+
"sliding_attention",
|
| 107 |
+
"sliding_attention",
|
| 108 |
+
"sliding_attention",
|
| 109 |
+
"full_attention",
|
| 110 |
+
"sliding_attention",
|
| 111 |
+
"sliding_attention",
|
| 112 |
+
"sliding_attention",
|
| 113 |
+
"sliding_attention",
|
| 114 |
+
"full_attention",
|
| 115 |
+
"sliding_attention",
|
| 116 |
+
"sliding_attention",
|
| 117 |
+
"sliding_attention",
|
| 118 |
+
"sliding_attention",
|
| 119 |
+
"full_attention",
|
| 120 |
+
"sliding_attention",
|
| 121 |
+
"sliding_attention",
|
| 122 |
+
"sliding_attention",
|
| 123 |
+
"sliding_attention",
|
| 124 |
+
"full_attention",
|
| 125 |
+
"sliding_attention",
|
| 126 |
+
"sliding_attention",
|
| 127 |
+
"sliding_attention",
|
| 128 |
+
"sliding_attention",
|
| 129 |
+
"full_attention",
|
| 130 |
+
"sliding_attention",
|
| 131 |
+
"sliding_attention",
|
| 132 |
+
"sliding_attention",
|
| 133 |
+
"sliding_attention",
|
| 134 |
+
"full_attention"
|
| 135 |
+
],
|
| 136 |
+
"max_position_embeddings": 131072,
|
| 137 |
+
"model_type": "gemma4_text",
|
| 138 |
+
"moe_intermediate_size": null,
|
| 139 |
+
"num_attention_heads": 8,
|
| 140 |
+
"num_experts": null,
|
| 141 |
+
"num_global_key_value_heads": null,
|
| 142 |
+
"num_hidden_layers": 35,
|
| 143 |
+
"num_key_value_heads": 1,
|
| 144 |
+
"num_kv_shared_layers": 20,
|
| 145 |
+
"output_attentions": false,
|
| 146 |
+
"output_hidden_states": true,
|
| 147 |
+
"pad_token_id": 0,
|
| 148 |
+
"problem_type": null,
|
| 149 |
+
"return_dict": true,
|
| 150 |
+
"rms_norm_eps": 1e-06,
|
| 151 |
+
"rope_parameters": {
|
| 152 |
+
"full_attention": {
|
| 153 |
+
"partial_rotary_factor": 0.25,
|
| 154 |
+
"rope_theta": 1000000.0,
|
| 155 |
+
"rope_type": "proportional"
|
| 156 |
+
},
|
| 157 |
+
"sliding_attention": {
|
| 158 |
+
"rope_theta": 10000.0,
|
| 159 |
+
"rope_type": "default"
|
| 160 |
+
}
|
| 161 |
+
},
|
| 162 |
+
"sliding_window": 512,
|
| 163 |
+
"tie_word_embeddings": true,
|
| 164 |
+
"top_k_experts": null,
|
| 165 |
+
"use_bidirectional_attention": null,
|
| 166 |
+
"use_cache": false,
|
| 167 |
+
"use_double_wide_mlp": true,
|
| 168 |
+
"vocab_size": 262144,
|
| 169 |
+
"vocab_size_per_layer_input": 262144
|
| 170 |
+
},
|
| 171 |
+
"tie_word_embeddings": true,
|
| 172 |
+
"transformers_version": "5.13.0.dev0",
|
| 173 |
+
"video_token_id": 258884,
|
| 174 |
+
"vision_config": {
|
| 175 |
+
"_name_or_path": "",
|
| 176 |
+
"architectures": null,
|
| 177 |
+
"attention_bias": false,
|
| 178 |
+
"attention_dropout": 0.0,
|
| 179 |
+
"chunk_size_feed_forward": 0,
|
| 180 |
+
"default_output_length": 280,
|
| 181 |
+
"dtype": "bfloat16",
|
| 182 |
+
"global_head_dim": 64,
|
| 183 |
+
"head_dim": 64,
|
| 184 |
+
"hidden_activation": "gelu_pytorch_tanh",
|
| 185 |
+
"hidden_size": 768,
|
| 186 |
+
"id2label": {
|
| 187 |
+
"0": "LABEL_0",
|
| 188 |
+
"1": "LABEL_1"
|
| 189 |
+
},
|
| 190 |
+
"initializer_range": 0.02,
|
| 191 |
+
"intermediate_size": 3072,
|
| 192 |
+
"is_encoder_decoder": false,
|
| 193 |
+
"label2id": {
|
| 194 |
+
"LABEL_0": 0,
|
| 195 |
+
"LABEL_1": 1
|
| 196 |
+
},
|
| 197 |
+
"max_position_embeddings": 131072,
|
| 198 |
+
"model_type": "gemma4_vision",
|
| 199 |
+
"num_attention_heads": 12,
|
| 200 |
+
"num_hidden_layers": 16,
|
| 201 |
+
"num_key_value_heads": 12,
|
| 202 |
+
"output_attentions": false,
|
| 203 |
+
"output_hidden_states": false,
|
| 204 |
+
"patch_size": 16,
|
| 205 |
+
"pooling_kernel_size": 3,
|
| 206 |
+
"position_embedding_size": 10240,
|
| 207 |
+
"problem_type": null,
|
| 208 |
+
"return_dict": true,
|
| 209 |
+
"rms_norm_eps": 1e-06,
|
| 210 |
+
"rope_parameters": {
|
| 211 |
+
"rope_theta": 100.0,
|
| 212 |
+
"rope_type": "default"
|
| 213 |
+
},
|
| 214 |
+
"standardize": false,
|
| 215 |
+
"use_clipped_linears": true
|
| 216 |
+
},
|
| 217 |
+
"vision_soft_tokens_per_image": 280
|
| 218 |
+
}
|
ckpt/epoch_0_step_3499/model/.hf_metadata/fqn_to_dtype_mapping.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
ckpt/epoch_0_step_3499/model/.hf_metadata/fqn_to_file_index_mapping.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
ckpt/epoch_0_step_3499/model/.hf_metadata/generation_config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_from_model_config": true,
|
| 3 |
+
"bos_token_id": 2,
|
| 4 |
+
"eos_token_id": 1,
|
| 5 |
+
"output_attentions": false,
|
| 6 |
+
"output_hidden_states": false,
|
| 7 |
+
"pad_token_id": 0,
|
| 8 |
+
"transformers_version": "5.13.0.dev0",
|
| 9 |
+
"use_cache": false
|
| 10 |
+
}
|
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:12bac982b793c44b03d52a250a9f0d0b666813da566b910c24a6da0695fd11e6
|
| 3 |
+
size 32170070
|
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json.ok
ADDED
|
File without changes
|
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer_config.json
ADDED
|
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_bos_token": true,
|
| 3 |
+
"audio_token": "<|audio|>",
|
| 4 |
+
"backend": "tokenizers",
|
| 5 |
+
"boa_token": "<|audio>",
|
| 6 |
+
"boi_token": "<|image>",
|
| 7 |
+
"bos_token": "<bos>",
|
| 8 |
+
"eoa_token": "<audio|>",
|
| 9 |
+
"eoc_token": "<channel|>",
|
| 10 |
+
"eoi_token": "<image|>",
|
| 11 |
+
"eos_token": "<eos>",
|
| 12 |
+
"eot_token": "<turn|>",
|
| 13 |
+
"escape_token": "<|\"|>",
|
| 14 |
+
"etc_token": "<tool_call|>",
|
| 15 |
+
"etd_token": "<tool|>",
|
| 16 |
+
"etr_token": "<tool_response|>",
|
| 17 |
+
"extra_special_tokens": [
|
| 18 |
+
"<|video|>"
|
| 19 |
+
],
|
| 20 |
+
"image_token": "<|image|>",
|
| 21 |
+
"mask_token": "<mask>",
|
| 22 |
+
"model_max_length": 1000000000000000019884624838656,
|
| 23 |
+
"pad_token": "<pad>",
|
| 24 |
+
"padding_side": "left",
|
| 25 |
+
"processor_class": "Gemma4Processor",
|
| 26 |
+
"soc_token": "<|channel>",
|
| 27 |
+
"sot_token": "<|turn>",
|
| 28 |
+
"stc_token": "<|tool_call>",
|
| 29 |
+
"std_token": "<|tool>",
|
| 30 |
+
"str_token": "<|tool_response>",
|
| 31 |
+
"think_token": "<|think|>",
|
| 32 |
+
"tokenizer_class": "GemmaTokenizer",
|
| 33 |
+
"unk_token": "<unk>"
|
| 34 |
+
}
|
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer_config.json.ok
ADDED
|
File without changes
|
ckpt/epoch_0_step_3499/model/consolidate.sh
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env bash
|
| 2 |
+
set -euo pipefail
|
| 3 |
+
|
| 4 |
+
# Offline HF safetensors consolidation helper.
|
| 5 |
+
# Defaults are conservative for login nodes and small CPU machines. For large
|
| 6 |
+
# checkpoints, run this on a CPU compute node and increase parallelism. Work is
|
| 7 |
+
# split across NPROC_PER_NODE worker processes, and each process uses NUM_THREADS
|
| 8 |
+
# writer threads; keep NPROC_PER_NODE * NUM_THREADS within your CPU allocation.
|
| 9 |
+
# Example for an 80-core CPU node:
|
| 10 |
+
# NPROC_PER_NODE=16 NUM_THREADS=5 bash "$0"
|
| 11 |
+
# Slurm example:
|
| 12 |
+
# sbatch --cpus-per-task=80 --wrap='NPROC_PER_NODE=16 NUM_THREADS=5 bash /path/to/consolidate.sh'
|
| 13 |
+
# Optional: set CAST_DTYPE=bf16 to request a floating-point dtype cast during export.
|
| 14 |
+
NPROC_PER_NODE="${NPROC_PER_NODE:-1}"
|
| 15 |
+
NUM_THREADS="${NUM_THREADS:-5}"
|
| 16 |
+
CAST_DTYPE="${CAST_DTYPE:-}"
|
| 17 |
+
PYTHON="${PYTHON:-python3}"
|
| 18 |
+
TORCHRUN="${TORCHRUN:-torchrun}"
|
| 19 |
+
CONSOLIDATION_TOOL="${CONSOLIDATION_TOOL:-tools/offline_hf_consolidation.py}"
|
| 20 |
+
CAST_DTYPE_ARGS=()
|
| 21 |
+
if [[ -n "${CAST_DTYPE}" ]]; then
|
| 22 |
+
CAST_DTYPE_ARGS=(--cast-dtype "${CAST_DTYPE}")
|
| 23 |
+
fi
|
| 24 |
+
if [[ ! -f "${CONSOLIDATION_TOOL}" ]]; then
|
| 25 |
+
echo "Could not find offline consolidation tool at ${CONSOLIDATION_TOOL}." >&2
|
| 26 |
+
echo "Run from the AutoModel repo root or set CONSOLIDATION_TOOL=/path/to/tools/offline_hf_consolidation.py." >&2
|
| 27 |
+
exit 1
|
| 28 |
+
fi
|
| 29 |
+
|
| 30 |
+
if [[ "${NPROC_PER_NODE}" -gt 1 ]]; then
|
| 31 |
+
"${TORCHRUN}" --nproc-per-node="${NPROC_PER_NODE}" "${CONSOLIDATION_TOOL}" \
|
| 32 |
+
--backend gloo \
|
| 33 |
+
--num-threads "${NUM_THREADS}" \
|
| 34 |
+
--model-name "/llm-align/open_models/gemma4/gemma-4-E2B" \
|
| 35 |
+
--input-dir "/llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt/epoch_0_step_3499/model" \
|
| 36 |
+
--output-dir "/llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt/epoch_0_step_3499/model/consolidated" \
|
| 37 |
+
"${CAST_DTYPE_ARGS[@]}"
|
| 38 |
+
else
|
| 39 |
+
"${PYTHON}" "${CONSOLIDATION_TOOL}" \
|
| 40 |
+
--backend gloo \
|
| 41 |
+
--num-threads "${NUM_THREADS}" \
|
| 42 |
+
--model-name "/llm-align/open_models/gemma4/gemma-4-E2B" \
|
| 43 |
+
--input-dir "/llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt/epoch_0_step_3499/model" \
|
| 44 |
+
--output-dir "/llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt/epoch_0_step_3499/model/consolidated" \
|
| 45 |
+
"${CAST_DTYPE_ARGS[@]}"
|
| 46 |
+
fi
|
ckpt/epoch_0_step_3499/model/shard-00001-model-00001-of-00001.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:262fb724221aed01d88de460cd159fc4255a824340dda830e32de7ec9fafe47e
|
| 3 |
+
size 1288086744
|
ckpt/epoch_0_step_3499/model/shard-00002-model-00001-of-00001.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0b96dfff86c554b6164a7e1b29df46f2fc49e018aa6a66f6e5b105abf66159c3
|
| 3 |
+
size 1288116916
|
ckpt/epoch_0_step_3499/model/shard-00003-model-00001-of-00001.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:775b23824afa1e21f3dce824d4bb95dc5014b4e153d282b2ebee8f53c3a53b07
|
| 3 |
+
size 1272388292
|
ckpt/epoch_0_step_3499/model/shard-00004-model-00001-of-00001.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6eacb735ec4d4fb3ca290cf4831720807db45fba746afbcf6f0b95fb6d7c5986
|
| 3 |
+
size 1272388220
|
ckpt/epoch_0_step_3499/model/shard-00005-model-00001-of-00001.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6fe94ed1afccacc48796118f6b77afa14366884771b958dce0970ebc863c2d22
|
| 3 |
+
size 1272388428
|
ckpt/epoch_0_step_3499/model/shard-00006-model-00001-of-00001.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0354e8a877afb40daa14291e9fe80cc4ae52fd4395cda664b9d9ce6c39803043
|
| 3 |
+
size 1272388450
|
ckpt/epoch_0_step_3499/model/shard-00007-model-00001-of-00001.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:81717e28af25472e36acff2dd53042f51c34d432a62a870feee53e012d392bce
|
| 3 |
+
size 1272388522
|
ckpt/epoch_0_step_3499/model/shard-00008-model-00001-of-00001.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:17ee142a7cee1aac26448d776cd5246bd072531ca7983376a8accf7892c995c0
|
| 3 |
+
size 1272388522
|
ckpt/epoch_0_step_3499/optim/.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0c30f0e0f383a0612f134105ff12ee0428bbf14ac74e537e147cd7cee8095238
|
| 3 |
+
size 4120680
|
ckpt/epoch_0_step_3499/optim/__0_0.distcp
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e8b6326eb3cbf4141b2533ae6095d5b4cc4632093860f44df4ba0cd097b82c7a
|
| 3 |
+
size 2348762093
|
ckpt/epoch_0_step_3499/optim/__1_0.distcp
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7ea21d2bcc982ed2437bfde6ddddf2d3b55636d353b01ed7cbea78558e53085c
|
| 3 |
+
size 2349308106
|
ckpt/epoch_0_step_3499/optim/__2_0.distcp
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:31522b1c00fbfea2a14c7b27bc820dcd85278c9eb1f49da691aaa826f524f792
|
| 3 |
+
size 2349308106
|
ckpt/epoch_0_step_3499/optim/__3_0.distcp
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5c4a568d75e66c0a537ace45bfa69201adc3c965df1f6ccd9708b6b8ef903773
|
| 3 |
+
size 2349308106
|
ckpt/epoch_0_step_3499/optim/__4_0.distcp
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ec6bc5c24b86c5952c4f24d4ccc265dc41b513c06dd48394f072300f65d8285
|
| 3 |
+
size 2349308106
|
ckpt/epoch_0_step_3499/optim/__5_0.distcp
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:13d61a702295622c336856dfbf58ed5d4a1e92bdcfd1d78fdfcd4b7336ba5314
|
| 3 |
+
size 2349306845
|
ckpt/epoch_0_step_3499/optim/__6_0.distcp
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:85e025abde21706f5bc74a3f52d98c9f8308b11db19e58adf829d2f021191b74
|
| 3 |
+
size 2349306845
|
ckpt/epoch_0_step_3499/optim/__7_0.distcp
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:449d329111eb4b6a1d39fcb6b141c45b72a638656be1747782ecc07b42c6ef1d
|
| 3 |
+
size 2349306845
|
ckpt/epoch_0_step_3499/rng/rng_global_rank_0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:127b09eac361efbf704e21cf1bee26f3100b64e0cf8b66694ab3eeb27b539d16
|
| 3 |
+
size 15461
|
ckpt/epoch_0_step_3499/rng/rng_global_rank_1.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf0a2ec856d02b763b9e0e3c2afffa721eeec648cbc0314079098707675124ef
|
| 3 |
+
size 15461
|
ckpt/epoch_0_step_3499/rng/rng_global_rank_2.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:953f7e48cebd0fe7ca90ac7ca4ded21d9226081fee24e12a2e362613a6ab3d62
|
| 3 |
+
size 15461
|
ckpt/epoch_0_step_3499/rng/rng_global_rank_3.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:833474b12f6e4ef051ccb0a07150a93f21f474804ada10df4255573bfd575176
|
| 3 |
+
size 15461
|
ckpt/epoch_0_step_3499/rng/rng_global_rank_4.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c1b60a138b1bdba4a6f08f6d34844db415a8a6eb01f336e813ec1206178818d8
|
| 3 |
+
size 15461
|
ckpt/epoch_0_step_3499/rng/rng_global_rank_5.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e12478054a12f4574601fbcc93bb4c2b18436d4ab957f97206a637fcff0e7c39
|
| 3 |
+
size 15461
|
ckpt/epoch_0_step_3499/rng/rng_global_rank_6.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:72e9cb72ea73bb6aa0eb17f870345ff4d13de92da878e00a08a5b0f8642b3f66
|
| 3 |
+
size 15461
|
ckpt/epoch_0_step_3499/rng/rng_global_rank_7.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e621f9ee1b9c83de09e0e417b240330d784eb81fb25b837cc69ea5999265b991
|
| 3 |
+
size 15461
|
ckpt/epoch_0_step_3499/step_scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8fe7446b9947b6c83210ae9ed2adb79a86ca9d494222a97052f2bbb6c4033247
|
| 3 |
+
size 1303
|