Views
No views yet
0.4.01base_model: llama-lang-adapt/pretrain-wura
2model_type: LlamaForCausalLM
3tokenizer_type: LlamaTokenizer
4is_llama_derived_model: true
5
6load_in_8bit: true
7load_in_4bit: false
8strict: false
9
10datasets:
11 - path: llama-lang-adapt/african-it
12 type: alpaca
13 train_on_split: train
14dataset_prepared_path: data/prepared-african-it
15
16test_datasets:
17 - path: llama-lang-adapt/african-it
18 type: alpaca
19 split: validation
20
21output_dir: ./lora-out
22
23sequence_len: 4096
24sample_packing: true
25pad_to_sequence_len: true
26
27adapter: lora
28lora_model_dir:
29lora_r: 32
30lora_alpha: 16
31lora_dropout: 0.05
32lora_target_linear: true
33lora_fan_in_fan_out:
34
35wandb_project:
36wandb_entity:
37wandb_watch:
38wandb_name:
39wandb_log_model:
40
41gradient_accumulation_steps: 8
42micro_batch_size: 1
43num_epochs: 1
44optimizer: adamw_bnb_8bit
45lr_scheduler: cosine
46learning_rate: 0.00002
47
48train_on_inputs: false
49group_by_length: false
50bf16: auto
51fp16: false
52tf32: false
53
54gradient_checkpointing: true
55early_stopping_patience:
56resume_from_checkpoint:
57local_rank:
58logging_steps: 1
59xformers_attention:
60flash_attention: true
61s2_attention:
62
63warmup_steps: 100
64evals_per_epoch: 4
65eval_table_size:
66saves_per_epoch: 1
67debug:
68weight_decay: 0.01
69fsdp:
70fsdp_config:
71special_tokens:
72| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.9958 | 0.0 | 1 | 3.1722 |
| 1.2509 | 0.25 | 7822 | 0.5396 |
| 1.0996 | 0.5 | 15644 | 0.5335 |
| 1.0109 | 0.75 | 23466 | 0.5321 |
| 1.0528 | 1.0 | 31288 | 0.5325 |