Views
No views yet
0.4.11base_model: openlm-research/open_llama_3b_v2
2model_type: LlamaForCausalLM
3tokenizer_type: LlamaTokenizer
4load_in_8bit: true
5load_in_4bit: false
6strict: false
7push_dataset_to_hub:
8datasets:
9 - path: teknium/GPT4-LLM-Cleaned
10 type: alpaca
11dataset_prepared_path:
12val_set_size: 0.02
13adapter: lora
14lora_model_dir:
15sequence_len: 1024
16sample_packing: true
17lora_r: 8
18lora_alpha: 16
19lora_dropout: 0.0
20lora_target_modules:
21 - gate_proj
22 - down_proj
23 - up_proj
24 - q_proj
25 - v_proj
26 - k_proj
27 - o_proj
28lora_fan_in_fan_out:
29wandb_project:
30wandb_entity:
31wandb_watch:
32wandb_name:
33wandb_log_model:
34output_dir: ./outputs/lora-out
35gradient_accumulation_steps: 1
36micro_batch_size: 2
37num_epochs: 2
38optimizer: adamw_bnb_8bit
39torchdistx_path:
40lr_scheduler: cosine
41learning_rate: 0.0002
42train_on_inputs: false
43group_by_length: false
44bf16: false
45fp16: true
46tf32: false
47gradient_checkpointing: true
48early_stopping_patience:
49resume_from_checkpoint:
50local_rank:
51logging_steps: 1
52xformers_attention:
53flash_attention: true
54gptq_groupsize:
55s2_attention:
56gptq_model_v1:
57warmup_steps: 20
58evals_per_epoch: 4
59saves_per_epoch: 1
60debug:
61deepspeed:
62weight_decay: 0.1
63fsdp:
64fsdp_config:
65special_tokens:
66 bos_token: "<s>"
67 eos_token: "</s>"
68 unk_token: "<unk>"
69| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.4057 | 0.0002 | 1 | 1.6558 |
| 0.9473 | 0.2501 | 1346 | 1.0867 |
| 1.0863 | 0.5003 | 2692 | 1.0618 |
| 0.827 | 0.7504 | 4038 | 1.0283 |
| 0.9852 | 1.0006 | 5384 | 1.0181 |
| 0.9818 | 1.2334 | 6730 | 1.0102 |
| 1.0796 | 1.4836 | 8076 | 1.0074 |
| 0.9089 | 1.7337 | 9422 | 1.0046 |