Views
No views yet
0.4.01base_model: mistralai/Mistral-7B-Instruct-v0.2
2model_type: AutoModelForCausalLM
3tokenizer_type: LlamaTokenizer
4is_mistral_derived_model: true
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8chat_template: inst
9
10datasets:
11 - path: ./data/nohto/training.jsonl
12 type: sharegpt
13
14dataset_prepared_path: last_run_prepared
15val_set_size: 0.1
16output_dir: ../nohto-v0-10e
17
18adapter: lora
19lora_model_dir:
20
21sequence_len: 4096
22sample_packing: true
23pad_to_sequence_len: true
24
25lora_r: 16
26lora_alpha: 32
27lora_dropout: 0.1
28lora_target_linear: true
29lora_fan_in_fan_out:
30
31eval_sample_packing: false
32
33hub_model_id: dyang415/nohto-v0-10e
34
35wandb_project: nohto
36wandb_name: nohto-v0
37wandb_log_model: end
38
39gradient_accumulation_steps: 2
40micro_batch_size: 1
41num_epochs: 10
42optimizer: paged_adamw_8bit
43lr_scheduler: cosine
44learning_rate: 0.0002
45
46train_on_inputs: false
47group_by_length: false
48bf16: auto
49fp16:
50tf32: false
51
52gradient_checkpointing: true
53early_stopping_patience:
54resume_from_checkpoint:
55local_rank:
56logging_steps: 1
57xformers_attention:
58flash_attention: true
59
60warmup_steps: 10
61eval_steps: 0.2
62save_steps: 0.1
63eval_max_new_tokens: 128
64debug:
65deepspeed:
66weight_decay: 0.0
67fsdp:
68fsdp_config:
69| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.7166 | 0.18 | 1 | 3.7658 |
| 0.5158 | 1.64 | 10 | 0.5278 |
| 0.2492 | 3.09 | 20 | 0.5739 |
| 0.0338 | 4.73 | 30 | 0.7476 |
| 0.0083 | 6.36 | 40 | 0.8089 |
| 0.0078 | 8.0 | 50 | 0.8229 |