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_torch_fused
8lr_scheduler: cosine
9learning_rate: 0.0001
10load_in_8bit: false
11load_in_4bit: true
12adapter: lora
13lora_model_dir: null
14lora_r: 4
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_4bits_r4
33peft_use_dora: false
34sequence_len: 4096
35sample_packing: false
36pad_to_sequence_len: false
37wandb_project: finlora_models
38wandb_entity: null
39wandb_watch: gradients
40wandb_name: formula_llama_3_1_8b_4bits_r4
41wandb_log_model: 'false'
42bf16: auto
43tf32: false
44gradient_checkpointing: true
45resume_from_checkpoint: null
46logging_steps: 500
47flash_attention: false
48deepspeed: deepspeed_configs/zero1.json
49warmup_steps: 10
50evals_per_epoch: 4
51saves_per_epoch: 1
52weight_decay: 0.0
53special_tokens:
54 pad_token: <|end_of_text|>
55chat_template: llama3
56| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.05 | 1 | 3.8659 |
| No log | 0.25 | 5 | 3.6317 |
| No log | 0.5 | 10 | 2.6735 |
| No log | 0.75 | 15 | 1.7570 |
| No log | 1.0 | 20 | 1.6143 |