Views
No views yet
0.11.0.dev01base_model: NousResearch/Meta-Llama-3-8B
2# optionally might have model_type or tokenizer_type
3model_type: LlamaForCausalLM
4tokenizer_type: AutoTokenizer
5# Automatically upload checkpoint and final model to HF
6# hub_model_id: username/custom_model_name
7
8load_in_8bit: true
9load_in_4bit: false
10
11datasets:
12 - path: cfierro/fr_alpaca
13 type: alpaca
14dataset_prepared_path: /workspace/axolotl-datasets/llama3-fr_alpaca
15val_set_size: 0.05
16output_dir: /workspace/axolotl-outputs/llama3-fr_alapaca-lora
17
18sequence_len: 4096
19sample_packing: true
20eval_sample_packing: false
21pad_to_sequence_len: true
22
23adapter: lora
24lora_model_dir:
25lora_r: 32
26lora_alpha: 16
27lora_dropout: 0.05
28lora_target_linear: true
29lora_modules_to_save:
30 - embed_tokens
31 - lm_head
32
33wandb_project: weight-diff-ft
34wandb_entity: cfierro
35wandb_watch: all
36wandb_name: llama3-fr_alpaca-lora
37wandb_log_model: "false"
38hub_model_id: cfierro/llama3-fr_alapaca-lora
39
40gradient_accumulation_steps: 4
41micro_batch_size: 2
42max_steps: 1000 # 8k examples at most
43optimizer: adamw_bnb_8bit
44lr_scheduler: cosine
45learning_rate: 0.0002
46
47bf16: auto
48tf32: false
49
50gradient_checkpointing: true
51resume_from_checkpoint:
52logging_steps: 1
53flash_attention: true
54
55warmup_steps: 10
56early_stopping_patience: 5
57eval_steps: 60 # bs=4*2 -> eval every 8*60=480 examples
58save_steps: 60 # needed for config validation
59save_total_limit: 1
60load_best_model_at_end: true
61weight_decay: 0.0
62special_tokens:
63 pad_token: <|end_of_text|>
64| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0 | 0 | 1.0157 |
| 0.8517 | 0.3061 | 60 | 0.8213 |
| 0.7548 | 0.6122 | 120 | 0.7892 |
| 0.7335 | 0.9184 | 180 | 0.7639 |
| 0.553 | 1.2245 | 240 | 0.7702 |
| 0.5687 | 1.5306 | 300 | 0.7620 |
| 0.6222 | 1.8367 | 360 | 0.7538 |
| 0.4544 | 2.1429 | 420 | 0.7803 |
| 0.4577 | 2.4490 | 480 | 0.7835 |
| 0.4629 | 2.7551 | 540 | 0.7733 |
| 0.3282 | 3.0612 | 600 | 0.8035 |
| 0.3396 | 3.3673 | 660 | 0.8144 |