Views
No views yet
0.3.01base_model: NousResearch/Llama-2-7b-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.0748 | 0.0 | 1 | 1.1154 |
| 0.8635 | 0.25 | 176 | 0.8732 |
| 0.8284 | 0.5 | 352 | 0.8463 |
| 0.7928 | 0.75 | 528 | 0.8295 |
| 0.8313 | 1.0 | 704 | 0.8155 |
| 0.6694 | 1.23 | 880 | 0.8196 |
| 0.636 | 1.48 | 1056 | 0.8144 |
| 0.6842 | 1.73 | 1232 | 0.8105 |
| 0.6277 | 1.98 | 1408 | 0.8100 |
bitsandbytes quantization config was used during training: