Views
No views yet
0.4.01base_model: meta-llama/Meta-Llama-3-8B
2model_type: LlamaForCausalLM
3tokenizer_type: AutoTokenizer
4strict: false
5
6# dataset
7datasets:
8 - path: BEE-spoke-data/bees-internal
9 type: completion # format from earlier
10 field: text # Optional[str] default: text, field to use for completion data
11val_set_size: 0.05
12
13sequence_len: 8192
14sample_packing: true
15pad_to_sequence_len: true
16train_on_inputs: false
17group_by_length: false
18
19# WANDB
20wandb_project: llama3-8bee
21wandb_entity: pszemraj
22wandb_watch: gradients
23wandb_name: llama3-8bee-8192
24hub_model_id: pszemraj/Meta-Llama-3-8Bee
25hub_strategy: every_save
26
27gradient_accumulation_steps: 8
28micro_batch_size: 1
29num_epochs: 1
30optimizer: paged_adamw_32bit
31lr_scheduler: cosine
32learning_rate: 2e-5
33
34load_in_8bit: false
35load_in_4bit: false
36bf16: auto
37fp16:
38tf32: true
39
40torch_compile: true # requires >= torch 2.0, may sometimes cause problems
41torch_compile_backend: inductor # Optional[str]
42gradient_checkpointing: true
43gradient_checkpointing_kwargs:
44 use_reentrant: false
45early_stopping_patience:
46logging_steps: 10
47xformers_attention:
48flash_attention: true
49
50warmup_steps: 25
51# hyperparams for freq of evals, saving, etc
52evals_per_epoch: 3
53saves_per_epoch: 3
54save_safetensors: true
55save_total_limit: 1 # Checkpoints saved at a time
56output_dir: ./output-axolotl/output-model-gamma
57resume_from_checkpoint:
58
59
60deepspeed:
61weight_decay: 0.0
62
63special_tokens:
64 pad_token: <|end_of_text|>BEE-spoke-data/bees-internal dataset (continued pretraining).
It achieves the following results on the evaluation set:| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0 | 1 | 2.5339 |
| 2.3719 | 0.33 | 232 | 2.3658 |
| 2.2914 | 0.67 | 464 | 2.3319 |