Views
No views yet
0.4.01base_model: mistralai/Mistral-7B-v0.3
2model_type: MistralForCausalLM
3tokenizer_type: LlamaTokenizer
4
5load_in_8bit: true
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: thewimo/german-medical-identification-dataset-v0.1
11 type: alpaca
12dataset_prepared_path: last_run_prepared
13val_set_size: 0.2
14output_dir: ./outputs/lora-out
15hub_model_id: thewimo/Mistral-7B-v0.3-deide-phi
16
17adapter: lora
18lora_model_dir:
19
20sequence_len: 4096
21sample_packing: false
22pad_to_sequence_len: true
23
24lora_r: 8
25lora_alpha: 16
26lora_dropout: 0.05
27lora_target_linear: true
28lora_fan_in_fan_out:
29lora_target_modules:
30 - gate_proj
31 - down_proj
32 - up_proj
33 - q_proj
34 - v_proj
35 - k_proj
36 - o_proj
37
38wandb_project: axolotl-runs
39wandb_entity: thewind-mom-finetuning
40wandb_watch:
41wandb_name: Mistral-7B-v0.3-deide-phi
42wandb_log_model:
43
44gradient_accumulation_steps: 4
45micro_batch_size: 4
46num_epochs: 4
47optimizer: paged_adamw_8bit
48lr_scheduler: cosine
49learning_rate: 0.0002
50
51train_on_inputs: false
52group_by_length: false
53bf16: auto
54fp16:
55tf32: false
56
57gradient_checkpointing: true
58early_stopping_patience:
59resume_from_checkpoint:
60local_rank:
61logging_steps: 1
62xformers_attention:
63flash_attention: true
64
65loss_watchdog_threshold: 5.0
66loss_watchdog_patience: 3
67
68warmup_steps: 10
69evals_per_epoch: 4
70eval_table_size:
71eval_max_new_tokens: 128
72saves_per_epoch: 1
73debug:
74deepspeed:
75weight_decay: 0.0
76fsdp:
77fsdp_config:
78special_tokens:
79| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.9682 | 0.0506 | 1 | 2.0579 |
| 1.2784 | 0.2532 | 5 | 0.8308 |
| 0.187 | 0.5063 | 10 | 0.1732 |
| 0.1094 | 0.7595 | 15 | 0.0819 |
| 0.0542 | 1.0127 | 20 | 0.0593 |
| 0.0354 | 1.2658 | 25 | 0.0521 |
| 0.0493 | 1.5190 | 30 | 0.0457 |
| 0.038 | 1.7722 | 35 | 0.0432 |
| 0.0143 | 2.0253 | 40 | 0.0425 |
| 0.0269 | 2.2785 | 45 | 0.0423 |
| 0.0273 | 2.5316 | 50 | 0.0415 |
| 0.0277 | 2.7848 | 55 | 0.0366 |
| 0.0288 | 3.0380 | 60 | 0.0356 |
| 0.0241 | 3.2911 | 65 | 0.0358 |
| 0.0125 | 3.5443 | 70 | 0.0362 |
| 0.0164 | 3.7975 | 75 | 0.0364 |