Views
No views yet
0.3.01base_model: out/
2model_type: MistralForCausalLM
3tokenizer_type: LlamaTokenizer
4is_mistral_derived_model: true
5
6load_in_8bit: false
7load_in_4bit: false
8strict: false
9
10datasets:
11 - path: cognitivecomputations/leet10k-alpaca
12 type: alpaca
13dataset_prepared_path:
14val_set_size: 0.05
15output_dir: ./out
16
17sequence_len: 8192
18sample_packing: true
19pad_to_sequence_len: true
20eval_sample_packing: false
21
22wandb_project:
23wandb_entity:
24wandb_watch:
25wandb_name:
26wandb_log_model:
27
28gradient_accumulation_steps: 4
29micro_batch_size: 2
30num_epochs: 1
31optimizer: adamw_bnb_8bit
32lr_scheduler: cosine
33learning_rate: 0.000005
34
35train_on_inputs: false
36group_by_length: false
37bf16: true
38fp16: false
39tf32: false
40
41gradient_checkpointing: true
42early_stopping_patience:
43resume_from_checkpoint:
44local_rank:
45logging_steps: 1
46xformers_attention:
47flash_attention: true
48
49warmup_steps: 10
50evals_per_epoch: 4
51eval_table_size:
52eval_table_max_new_tokens: 128
53saves_per_epoch: 1
54debug:
55deepspeed:
56weight_decay: 0.0
57fsdp:
58fsdp_config:
59special_tokens:
60 bos_token: "<s>"
61 eos_token: "</s>"
62 unk_token: "<unk>"
63| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.7842 | 0.01 | 1 | 0.8053 |
| 0.5057 | 0.26 | 35 | 0.5694 |
| 0.3987 | 0.51 | 70 | 0.5752 |
| 0.2964 | 0.77 | 105 | 0.5907 |
| Metric | Value |
|---|---|
| Avg. | 56.57 |
| AI2 Reasoning Challenge (25-Shot) | 61.01 |
| HellaSwag (10-Shot) | 84.12 |
| MMLU (5-Shot) | 64.05 |
| TruthfulQA (0-shot) | 42.21 |
| Winogrande (5-shot) | 79.87 |
| GSM8k (5-shot) | 8.19 |