Views
No views yet
0.9.11base_model: meta-llama/Llama-3.1-8B-Instruct
2model_type: LlamaForCausalLM
3tokenizer_type: AutoTokenizer
4gradient_accumulation_steps: 2
5micro_batch_size: 4
6num_epochs: 1
7optimizer: adamw_bnb_8bit
8lr_scheduler: cosine
9learning_rate: 0.0001
10load_in_8bit: true
11load_in_4bit: false
12adapter: lora
13lora_model_dir: null
14lora_r: 8
15lora_alpha: 16
16lora_dropout: 0.05
17lora_target_modules:
18- q_proj
19- v_proj
20- k_proj
21datasets:
22- path: /workspace/FinLoRA/data/train/formula_train.jsonl
23 type:
24 system_prompt: ''
25 field_system: system
26 field_instruction: context
27 field_output: target
28 format: '[INST] {instruction} [/INST]'
29 no_input_format: '[INST] {instruction} [/INST]'
30dataset_prepared_path: null
31val_set_size: 0.02
32output_dir: /workspace/FinLoRA/lora/axolotl-output/formula_llama_3_1_8b_8bits_r8_rslora
33peft_use_dora: false
34peft_use_rslora: true
35sequence_len: 4096
36sample_packing: false
37pad_to_sequence_len: false
38wandb_project: finlora_models
39wandb_entity: null
40wandb_watch: gradients
41wandb_name: formula_llama_3_1_8b_8bits_r8_rslora
42wandb_log_model: 'false'
43bf16: auto
44tf32: false
45gradient_checkpointing: true
46resume_from_checkpoint: null
47logging_steps: 500
48flash_attention: false
49deepspeed: deepspeed_configs/zero1.json
50warmup_steps: 10
51evals_per_epoch: 4
52saves_per_epoch: 1
53weight_decay: 0.0
54special_tokens:
55 pad_token: <|end_of_text|>
56chat_template: llama3
57| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.05 | 1 | 4.5176 |
| No log | 0.25 | 5 | 3.4286 |
| No log | 0.5 | 10 | 1.6787 |
| No log | 0.75 | 15 | 1.2956 |
| No log | 1.0 | 20 | 1.2273 |