Views
No views yet
0.3.01base_model: epfl-llm/meditron-7b
2model_type: LlamaForCausalLM
3tokenizer_type: LlamaTokenizer
4is_llama_derived_model: true
5
6load_in_8bit: false
7load_in_4bit: true
8strict: false
9
10datasets:
11 - path: Open-Orca/SlimOrca-Dedup
12 type: sharegpt
13 - path: axiong/pmc_llama_instructions
14 type: alpaca
15 - path: xzuyn/chatdoctor-200k-stripped
16 type: alpaca
17 - path: technoculture/riddle_sense
18 type: alpaca
19dataset_prepared_path:
20val_set_size: 0.05
21output_dir: ./qlora-out
22
23adapter: qlora
24lora_model_dir:
25
26sequence_len: 2048
27sample_packing: true
28pad_to_sequence_len: true
29
30lora_r: 32
31lora_alpha: 16
32lora_dropout: 0.05
33lora_target_modules:
34lora_target_linear: true
35lora_fan_in_fan_out:
36
37wandb_project: MD7b-alpha
38wandb_entity: technoculture
39wandb_watch:
40wandb_name:
41wandb_log_model: true
42
43gradient_accumulation_steps: 4
44micro_batch_size: 2
45num_epochs: 4
46optimizer: paged_adamw_32bit
47lr_scheduler_type: cosine
48lr_scheduler: cosine
49learning_rate: 0.0003
50
51train_on_inputs: false
52group_by_length: false
53bf16: true
54fp16: false
55tf32: false
56
57do_eval: true
58evals_per_epoch: 2
59eval_table_size:
60saves_per_epoch: 1
61
62hub_model_id: technoculture/md7b-alpha
63hub_strategy: every_save
64push_to_hub: true
65
66log_level: info
67logging_steps: 1
68logging_strategy: steps
69
70gradient_checkpointing: true
71early_stopping_patience:
72resume_from_checkpoint: false
73local_rank:
74xformers_attention:
75flash_attention: true
76
77warmup_steps: 2000
78debug:
79deepspeed:
80weight_decay: 0.1
81fsdp:
82fsdp_config:
83special_tokens:
84 bos_token: "<s>"
85 eos_token: "</s>"
86 unk_token: "<unk>"
87| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 2.1602 | 0.0 | 1 | 1.9066 |
| 1.1128 | 0.5 | 14744 | 1.1620 |
| 1.2463 | 1.0 | 29488 | 1.1288 |
| 0.8291 | 1.49 | 44232 | 1.1025 |
| 1.0524 | 1.99 | 58976 | 1.0771 |
| 1.0369 | 2.48 | 73720 | 1.0563 |
| 1.0402 | 2.98 | 88464 | 1.0299 |
| 0.943 | 3.47 | 103208 | 1.0271 |
| 1.0845 | 3.97 | 117952 | 1.0238 |
bitsandbytes quantization config was used during training: