Views
No views yet
0.4.11base_model: NousResearch/Meta-Llama-3.1-8B-Instruct
2model_type: LlamaForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: shivamsark/simjud-800
11 type: alpaca
12
13dataset_prepared_path:
14val_set_size: 0.05
15output_dir: ./outputs/lora-out
16
17sequence_len: 4096
18sample_packing: false
19pad_to_sequence_len: true
20
21adapter: lora
22lora_model_dir:
23lora_r: 32
24lora_alpha: 16
25lora_dropout: 0.05
26lora_target_linear: true
27lora_fan_in_fan_out:
28
29wandb_project:
30wandb_entity:
31wandb_watch:
32wandb_name:
33wandb_log_model:
34
35gradient_accumulation_steps: 4
36micro_batch_size: 2
37num_epochs: 1
38optimizer: adamw_bnb_8bit
39lr_scheduler: cosine
40learning_rate: 0.0002
41
42train_on_inputs: false
43group_by_length: false
44bf16: auto
45fp16:
46tf32: false
47
48gradient_checkpointing: true
49early_stopping_patience:
50resume_from_checkpoint:
51local_rank:
52logging_steps: 1
53xformers_attention:
54flash_attention: true
55s2_attention:
56
57warmup_steps: 10
58evals_per_epoch: 4
59eval_table_size:
60eval_max_new_tokens: 128
61saves_per_epoch: 1
62debug:
63deepspeed:
64weight_decay: 0.0
65fsdp:
66fsdp_config:
67special_tokens:
68 pad_token: <|end_of_text|>
69| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 10.1436 | 0.0104 | 1 | 10.4027 |
| 0.3719 | 0.2494 | 24 | 0.3511 |
| 0.0667 | 0.4987 | 48 | 0.3615 |
| 0.149 | 0.7481 | 72 | 0.0832 |
| 0.1657 | 0.9974 | 96 | 0.1033 |