Views
No views yet
0.3.01base_model: NousResearch/Llama-2-13b-hf
2model_type: LlamaForCausalLM
3tokenizer_type: LlamaTokenizer
4is_llama_derived_model: true
5
6load_in_8bit: true
7load_in_4bit: false
8strict: false
9
10datasets:
11 - path: dataset
12 type: sharegpt
13dataset_prepared_path:
14val_set_size: 0.05
15output_dir: ./lora-out
16
17sequence_len: 4096
18sample_packing: true
19pad_to_sequence_len: true
20
21adapter: lora
22lora_model_dir:
23lora_r: 128
24lora_alpha: 64
25lora_dropout: 0.05
26lora_target_linear: true
27lora_fan_in_fan_out:
28
29wandb_project: toxicLlama-2-13B
30wandb_entity:
31wandb_watch:
32wandb_name:
33wandb_log_model:
34
35gradient_accumulation_steps: 1
36micro_batch_size: 2
37num_epochs: 2
38optimizer: adamw_bnb_8bit
39lr_scheduler: cosine
40learning_rate: 0.0002
41eval_batch_size: 2
42
43train_on_inputs: false
44group_by_length: false
45bf16: true
46fp16: false
47tf32: false
48
49gradient_checkpointing: true
50early_stopping_patience:
51resume_from_checkpoint:
52local_rank:
53logging_steps: 1
54xformers_attention:
55flash_attention: true
56
57warmup_steps: 10
58evals_per_epoch: 4
59eval_table_size:
60eval_table_max_new_tokens: 128
61saves_per_epoch: 1
62debug:
63deepspeed:
64weight_decay: 0.0
65fsdp:
66fsdp_config:
67special_tokens:
68 bos_token: "<s>"
69 eos_token: "</s>"
70 unk_token: "<unk>"
71| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.0107 | 0.0 | 1 | 1.0286 |
| 0.8198 | 0.25 | 152 | 0.8079 |
| 0.7993 | 0.5 | 304 | 0.7904 |
| 0.7348 | 0.75 | 456 | 0.7748 |
| 0.689 | 1.0 | 608 | 0.7638 |
| 0.6462 | 1.23 | 760 | 0.7729 |
| 0.6226 | 1.48 | 912 | 0.7657 |
| 0.6179 | 1.73 | 1064 | 0.7634 |
bitsandbytes quantization config was used during training: