Views
No views yet
0.4.01base_model: meta-llama/Llama-2-7b-hf
2base_model_config: meta-llama/Llama-2-7b-hf
3model_type: LlamaForCausalLM
4tokenizer_type: LlamaTokenizer
5is_llama_derived_model: true
6
7load_in_8bit: false
8load_in_4bit: true
9strict: false
10
11datasets:
12 - path: ascherrer/mtext-data-150224_2
13 type: completion
14 field: text
15dataset_prepared_path: last_run_prepared
16hub_model_id: ascherrer/mtext-150224_2
17val_set_size: 0.01
18output_dir: ./qlora-out
19
20adapter: qlora
21lora_model_dir:
22
23
24sequence_len: 4096
25sample_packing: true
26eval_sample_packing: false
27
28pad_to_sequence_len: true
29
30lora_r: 32
31lora_alpha: 16
32lora_dropout: 0.05
33lora_target_modules:
34lora_target_linear: true
35lora_fan_in_fan_out:
36
37wandb_project: "machine-de-textes"
38wandb_entity:
39wandb_watch:
40wandb_run_id:
41wandb_log_model: "checkpoint"
42
43lora_modules_to_save:
44 - embed_tokens
45 - lm_head
46
47gradient_accumulation_steps: 4
48micro_batch_size: 2
49num_epochs: 3
50optimizer: paged_adamw_32bit
51lr_scheduler: cosine
52learning_rate: 0.0002
53
54train_on_inputs: false
55group_by_length: false
56bf16: true
57fp16: false
58tf32: false
59
60gradient_checkpointing: true
61early_stopping_patience:
62resume_from_checkpoint:
63local_rank:
64logging_steps: 1
65xformers_attention:
66flash_attention: true
67
68warmup_steps: 10
69eval_steps: 20
70eval_table_size: 5
71save_steps:
72debug:
73deepspeed:
74weight_decay: 0.0
75fsdp:
76fsdp_config:
77special_tokens:
78 bos_token: "<s>"
79 eos_token: "</s>"
80 unk_token: "<unk>"
81tokens: # these are delimiters
82 - "<|s|>"
83 - "<|e|>"| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.8407 | 0.42 | 20 | 2.1162 |
| 1.6743 | 0.84 | 40 | 2.0768 |
| 1.5006 | 1.24 | 60 | 2.0654 |
| 1.5812 | 1.65 | 80 | 2.0598 |
| 1.5619 | 2.05 | 100 | 2.0535 |
| 1.5251 | 2.47 | 120 | 2.0537 |
| 1.5473 | 2.89 | 140 | 2.0540 |