Views
No views yet
0.3.01base_model: /data/internlm2-base-20b-llama
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: /data/train-all-8k.jsonl
12 type: completion
13dataset_prepared_path:
14val_set_size: 0.05
15output_dir: /data/internlm-limarp-lora-out
16
17sequence_len: 8192
18sample_packing: true
19pad_to_sequence_len: true
20
21adapter: lora
22lora_model_dir:
23lora_r: 128
24lora_alpha: 64
25lora_dropout: 0.05
26lora_target_linear: true
27lora_fan_in_fan_out:
28
29gradient_accumulation_steps: 4
30micro_batch_size: 2
31num_epochs: 4
32optimizer: adamw_bnb_8bit
33lr_scheduler: cosine
34learning_rate: 0.00002
35
36train_on_inputs: false
37group_by_length: false
38bf16: true
39fp16: false
40tf32: false
41
42gradient_checkpointing: true
43gradient_checkpointing_kwargs:
44early_stopping_patience:
45resume_from_checkpoint:
46local_rank:
47logging_steps: 1
48xformers_attention:
49flash_attention: true
50s2_attention:
51
52warmup_steps: 10
53evals_per_epoch: 4
54eval_table_size:
55eval_table_max_new_tokens: 128
56saves_per_epoch: 1
57debug:
58deepspeed:
59weight_decay: 0.0
60fsdp:
61fsdp_config:
62special_tokens:
63 bos_token: "<s>"
64 eos_token: "</s>"
65 unk_token: "<unk>"
66bitsandbytes quantization config was used during training:| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 2.3563 | 0.01 | 1 | 2.3995 |
| 2.1815 | 0.25 | 37 | 2.2693 |
| 2.1364 | 0.51 | 74 | 2.1684 |
| 2.1355 | 0.76 | 111 | 2.1526 |
| 2.1624 | 1.03 | 148 | 2.1435 |
| 2.1326 | 1.28 | 185 | 2.1367 |
| 1.9987 | 1.54 | 222 | 2.1330 |
| 2.0494 | 1.79 | 259 | 2.1291 |
| 2.0505 | 2.04 | 296 | 2.1266 |
| 2.075 | 2.3 | 333 | 2.1243 |
| 2.0183 | 2.55 | 370 | 2.1229 |
| 2.1047 | 2.81 | 407 | 2.1227 |
| 2.1309 | 3.06 | 444 | 2.1218 |
| 2.1249 | 3.31 | 481 | 2.1214 |
| 2.1423 | 3.57 | 518 | 2.1214 |
| 2.0913 | 3.82 | 555 | 2.1216 |