Commit
•
59010b3
1
Parent(s):
869e25b
Delete config_tiny_mistral.yaml
Browse files- config_tiny_mistral.yaml +0 -90
config_tiny_mistral.yaml
DELETED
|
@@ -1,90 +0,0 @@
|
|
| 1 |
-
checkpoints:
|
| 2 |
-
checkpoint_interval: 10
|
| 3 |
-
checkpoints_path: /fsx/nouamane/projects/nanotron/checkpoints
|
| 4 |
-
checkpoints_path_is_shared_file_system: false
|
| 5 |
-
resume_checkpoint_path: null
|
| 6 |
-
save_initial_state: false
|
| 7 |
-
data:
|
| 8 |
-
dataset:
|
| 9 |
-
dataset_overwrite_cache: false
|
| 10 |
-
dataset_processing_num_proc_per_process: 1
|
| 11 |
-
hf_dataset_config_name: null
|
| 12 |
-
hf_dataset_or_datasets: HuggingFaceH4/testing_alpaca_small
|
| 13 |
-
hf_dataset_splits: train
|
| 14 |
-
text_column_name: completion
|
| 15 |
-
num_loading_workers: 1
|
| 16 |
-
seed: 42
|
| 17 |
-
general:
|
| 18 |
-
benchmark_csv_path: null
|
| 19 |
-
consumed_train_samples: null
|
| 20 |
-
ignore_sanity_checks: false
|
| 21 |
-
project: debug
|
| 22 |
-
run: tiny_mistral
|
| 23 |
-
seed: 42
|
| 24 |
-
step: null
|
| 25 |
-
logging:
|
| 26 |
-
iteration_step_info_interval: 1
|
| 27 |
-
log_level: info
|
| 28 |
-
log_level_replica: info
|
| 29 |
-
model:
|
| 30 |
-
ddp_bucket_cap_mb: 25
|
| 31 |
-
dtype: bfloat16
|
| 32 |
-
init_method:
|
| 33 |
-
std: 0.025
|
| 34 |
-
make_vocab_size_divisible_by: 1
|
| 35 |
-
model_config:
|
| 36 |
-
bos_token_id: 1
|
| 37 |
-
eos_token_id: 2
|
| 38 |
-
hidden_act: silu
|
| 39 |
-
hidden_size: 16
|
| 40 |
-
initializer_range: 0.02
|
| 41 |
-
intermediate_size: 64
|
| 42 |
-
is_mistral_config: true
|
| 43 |
-
max_position_embeddings: 256
|
| 44 |
-
num_attention_heads: 4
|
| 45 |
-
num_hidden_layers: 2
|
| 46 |
-
num_key_value_heads: 4
|
| 47 |
-
pad_token_id: null
|
| 48 |
-
pretraining_tp: 1
|
| 49 |
-
rms_norm_eps: 1.0e-05
|
| 50 |
-
rope_scaling: null
|
| 51 |
-
tie_word_embeddings: true
|
| 52 |
-
use_cache: true
|
| 53 |
-
vocab_size: 256
|
| 54 |
-
optimizer:
|
| 55 |
-
accumulate_grad_in_fp32: true
|
| 56 |
-
adam_beta1: 0.9
|
| 57 |
-
adam_beta2: 0.95
|
| 58 |
-
adam_eps: 1.0e-08
|
| 59 |
-
clip_grad: 1.0
|
| 60 |
-
learning_rate_scheduler:
|
| 61 |
-
learning_rate: 0.0003
|
| 62 |
-
lr_decay_steps: 8
|
| 63 |
-
lr_decay_style: cosine
|
| 64 |
-
lr_warmup_steps: 2
|
| 65 |
-
lr_warmup_style: linear
|
| 66 |
-
min_decay_lr: 1.0e-05
|
| 67 |
-
torch_adam_is_fused: true
|
| 68 |
-
weight_decay: 0.01
|
| 69 |
-
zero_stage: 0
|
| 70 |
-
parallelism:
|
| 71 |
-
dp: 2
|
| 72 |
-
pp: 2
|
| 73 |
-
pp_engine: 1f1b
|
| 74 |
-
recompute_granularity: SELECTIVE
|
| 75 |
-
tp: 2
|
| 76 |
-
tp_linear_async_communication: true
|
| 77 |
-
tp_mode: REDUCE_SCATTER
|
| 78 |
-
profiler: null
|
| 79 |
-
tokenizer:
|
| 80 |
-
tokenizer_max_length: null
|
| 81 |
-
tokenizer_name_or_path: gpt2
|
| 82 |
-
tokenizer_revision: null
|
| 83 |
-
tokens:
|
| 84 |
-
batch_accumulation_per_replica: 1
|
| 85 |
-
limit_test_batches: 0
|
| 86 |
-
limit_val_batches: 0
|
| 87 |
-
micro_batch_size: 2
|
| 88 |
-
sequence_length: 32
|
| 89 |
-
train_steps: 10
|
| 90 |
-
val_check_interval: -1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|