1base_model: meta-llama/Llama-3.2-1B-Instruct
2# Automatically upload checkpoint and final model to HF
3hub_model_id: syvai/no-emotion-reasoning-1b
4
5datasets:
6 - path: syvai/no-emotion-reasoning
7 type: chat_template
8
9dataset_prepared_path: last_run_prepared
10val_set_size: 0.02
11output_dir: ./outputs/out
12
13sequence_len: 8192
14sample_packing: true
15pad_to_sequence_len: true
16
17wandb_project: reasoning-emotions
18wandb_entity:
19wandb_watch:
20wandb_name:
21wandb_log_model:
22
23eval_sample_packing: False
24gradient_accumulation_steps: 8
25micro_batch_size: 2
26num_epochs: 1
27optimizer: paged_adamw_8bit
28lr_scheduler: cosine
29learning_rate: 2e-5
30
31bf16: auto
32tf32: false
33
34gradient_checkpointing: true
35gradient_checkpointing_kwargs:
36 use_reentrant: false
37resume_from_checkpoint:
38logging_steps: 1
39flash_attention: true
40
41warmup_steps: 1
42evals_per_epoch: 2
43saves_per_epoch: 1
44weight_decay: 0.0
45special_tokens:
46 pad_token: <|end_of_text|>
47