Views
No views yet
0.9.1.post11base_model: meta-llama/Llama-3.1-8B-Instruct
2model_type: LlamaForCausalLM
3tokenizer_type: AutoTokenizer
4gradient_accumulation_steps: 8
5micro_batch_size: 1
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/xbrl_extract_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/xbrl_extract_llama_3_1_8b_8bits_r8
33peft_use_dora: false
34peft_use_rslora: false
35sequence_len: 4096
36sample_packing: false
37pad_to_sequence_len: false
38wandb_project: finlora_models
39wandb_entity: null
40wandb_watch: gradients
41wandb_name: xbrl_extract_llama_3_1_8b_8bits_r8
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.0038 | 1 | 1.6299 |
| No log | 0.2526 | 67 | 0.0075 |
| No log | 0.5052 | 134 | 0.0037 |
| No log | 0.7578 | 201 | 0.0025 |