File size: 18,019 Bytes
bec5fb2 e6a8b5a 6798fe4 3ff731f 6798fe4 e97474f 4c3c48f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 |
---
license: apache-2.0
---
# tangled-llama-j-128k-v0.1
## Train Tokenizer
```bash
python -B train_tokenizer.py
```
Tokenizer training log:
```
Resolving data files: 100%|████████████████████████████████████████████████████████████████| 132/132 [00:00<00:00, 266.56it/s]
Loading dataset shards: 100%|█████████████████████████████████████████████████████████████████| 18/18 [00:05<00:00, 3.24it/s]
Resolving data files: 100%|█████████████████████████████████████████████████████████████| 133/133 [00:00<00:00, 306844.02it/s]
[00:21:52] Pre-processing sequences ████████████████████████████████████████████████████████████████ 0 / 0
[00:00:48] Tokenize words ████████████████████████████████████████████████████████████████ 25635525 / 25635525
[00:01:17] Count pairs ████████████████████████████████████████████████████████████████ 25635525 / 25635525
[00:06:07] Compute merges ████████████████████████████████████████████████████████████████ 32066 / 32066
```
## Pretrain
```bash
python -B prepare_pretrain_dataset.py
```
```bash
CUDA_VISIBLE_DEVICES=0 CUDA_LAUNCH_BLOCKING=0 PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True litgpt pretrain --config pretrain-model.yaml
```
## Chat with Pretrained model
```bash
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True CUDA_VISIBLE_DEVICES="0" litgpt chat out/pretrain/final/
```
## Evaluation
### Quick
```bash
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True CUDA_VISIBLE_DEVICES="0" litgpt evaluate --tasks 'hellaswag,gsm8k,truthfulqa_mc2,mmlu,winogrande,arc_challenge' --out_dir 'evaluate-quick/' --batch_size 8 --dtype 'bfloat16' out/pretrain/final/
```
```
| Tasks |Version| Filter |n-shot| Metric | |Value | |Stderr|
|---------------------------------------|------:|----------------|-----:|-----------|---|-----:|---|-----:|
|arc_challenge | 1|none | 0|acc |↑ |0.1715|± |0.0110|
| | |none | 0|acc_norm |↑ |0.2150|± |0.0120|
|gsm8k | 3|flexible-extract| 5|exact_match|↑ |0.0136|± |0.0032|
| | |strict-match | 5|exact_match|↑ |0.0114|± |0.0029|
|hellaswag | 1|none | 0|acc |↑ |0.2715|± |0.0044|
| | |none | 0|acc_norm |↑ |0.2819|± |0.0045|
|mmlu | 2|none | |acc |↑ |0.2307|± |0.0036|
| - humanities | 2|none | |acc |↑ |0.2436|± |0.0063|
| - formal_logic | 1|none | 0|acc |↑ |0.3175|± |0.0416|
| - high_school_european_history | 1|none | 0|acc |↑ |0.2606|± |0.0343|
| - high_school_us_history | 1|none | 0|acc |↑ |0.2598|± |0.0308|
| - high_school_world_history | 1|none | 0|acc |↑ |0.2700|± |0.0289|
| - international_law | 1|none | 0|acc |↑ |0.2479|± |0.0394|
| - jurisprudence | 1|none | 0|acc |↑ |0.2870|± |0.0437|
| - logical_fallacies | 1|none | 0|acc |↑ |0.2209|± |0.0326|
| - moral_disputes | 1|none | 0|acc |↑ |0.2457|± |0.0232|
| - moral_scenarios | 1|none | 0|acc |↑ |0.2380|± |0.0142|
| - philosophy | 1|none | 0|acc |↑ |0.1833|± |0.0220|
| - prehistory | 1|none | 0|acc |↑ |0.2160|± |0.0229|
| - professional_law | 1|none | 0|acc |↑ |0.2438|± |0.0110|
| - world_religions | 1|none | 0|acc |↑ |0.2924|± |0.0349|
| - other | 2|none | |acc |↑ |0.2385|± |0.0076|
| - business_ethics | 1|none | 0|acc |↑ |0.2900|± |0.0456|
| - clinical_knowledge | 1|none | 0|acc |↑ |0.2075|± |0.0250|
| - college_medicine | 1|none | 0|acc |↑ |0.2139|± |0.0313|
| - global_facts | 1|none | 0|acc |↑ |0.1800|± |0.0386|
| - human_aging | 1|none | 0|acc |↑ |0.3139|± |0.0311|
| - management | 1|none | 0|acc |↑ |0.1748|± |0.0376|
| - marketing | 1|none | 0|acc |↑ |0.2991|± |0.0300|
| - medical_genetics | 1|none | 0|acc |↑ |0.2800|± |0.0451|
| - miscellaneous | 1|none | 0|acc |↑ |0.2363|± |0.0152|
| - nutrition | 1|none | 0|acc |↑ |0.2157|± |0.0236|
| - professional_accounting | 1|none | 0|acc |↑ |0.2376|± |0.0254|
| - professional_medicine | 1|none | 0|acc |↑ |0.1838|± |0.0235|
| - virology | 1|none | 0|acc |↑ |0.2892|± |0.0353|
| - social sciences | 2|none | |acc |↑ |0.2181|± |0.0074|
| - econometrics | 1|none | 0|acc |↑ |0.2368|± |0.0400|
| - high_school_geography | 1|none | 0|acc |↑ |0.1768|± |0.0272|
| - high_school_government_and_politics| 1|none | 0|acc |↑ |0.2073|± |0.0293|
| - high_school_macroeconomics | 1|none | 0|acc |↑ |0.2103|± |0.0207|
| - high_school_microeconomics | 1|none | 0|acc |↑ |0.2101|± |0.0265|
| - high_school_psychology | 1|none | 0|acc |↑ |0.1927|± |0.0169|
| - human_sexuality | 1|none | 0|acc |↑ |0.2519|± |0.0381|
| - professional_psychology | 1|none | 0|acc |↑ |0.2533|± |0.0176|
| - public_relations | 1|none | 0|acc |↑ |0.2182|± |0.0396|
| - security_studies | 1|none | 0|acc |↑ |0.1796|± |0.0246|
| - sociology | 1|none | 0|acc |↑ |0.2438|± |0.0304|
| - us_foreign_policy | 1|none | 0|acc |↑ |0.2700|± |0.0446|
| - stem | 2|none | |acc |↑ |0.2160|± |0.0073|
| - abstract_algebra | 1|none | 0|acc |↑ |0.2000|± |0.0402|
| - anatomy | 1|none | 0|acc |↑ |0.1852|± |0.0336|
| - astronomy | 1|none | 0|acc |↑ |0.1776|± |0.0311|
| - college_biology | 1|none | 0|acc |↑ |0.2569|± |0.0365|
| - college_chemistry | 1|none | 0|acc |↑ |0.1900|± |0.0394|
| - college_computer_science | 1|none | 0|acc |↑ |0.2700|± |0.0446|
| - college_mathematics | 1|none | 0|acc |↑ |0.2200|± |0.0416|
| - college_physics | 1|none | 0|acc |↑ |0.2255|± |0.0416|
| - computer_security | 1|none | 0|acc |↑ |0.3000|± |0.0461|
| - conceptual_physics | 1|none | 0|acc |↑ |0.2638|± |0.0288|
| - electrical_engineering | 1|none | 0|acc |↑ |0.2276|± |0.0349|
| - elementary_mathematics | 1|none | 0|acc |↑ |0.2037|± |0.0207|
| - high_school_biology | 1|none | 0|acc |↑ |0.1903|± |0.0223|
| - high_school_chemistry | 1|none | 0|acc |↑ |0.1823|± |0.0272|
| - high_school_computer_science | 1|none | 0|acc |↑ |0.2500|± |0.0435|
| - high_school_mathematics | 1|none | 0|acc |↑ |0.2259|± |0.0255|
| - high_school_physics | 1|none | 0|acc |↑ |0.2119|± |0.0334|
| - high_school_statistics | 1|none | 0|acc |↑ |0.1574|± |0.0248|
| - machine_learning | 1|none | 0|acc |↑ |0.2768|± |0.0425|
|truthfulqa_mc2 | 2|none | 0|acc |↑ |0.4649|± |0.0155|
|winogrande | 1|none | 0|acc |↑ |0.4988|± |0.0141|
| Groups |Version|Filter|n-shot|Metric| |Value | |Stderr|
|------------------|------:|------|------|------|---|-----:|---|-----:|
|mmlu | 2|none | |acc |↑ |0.2307|± |0.0036|
| - humanities | 2|none | |acc |↑ |0.2436|± |0.0063|
| - other | 2|none | |acc |↑ |0.2385|± |0.0076|
| - social sciences| 2|none | |acc |↑ |0.2181|± |0.0074|
| - stem | 2|none | |acc |↑ |0.2160|± |0.0073|
```
### Leaderboard
```bash
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True CUDA_VISIBLE_DEVICES="0" litgpt evaluate --tasks 'leaderboard' --out_dir 'evaluate-leaderboard/' --batch_size 8 --dtype 'bfloat16' out/pretrain/final/
```
```
| Tasks |Version|Filter|n-shot| Metric | |Value | |Stderr|
|-----------------------------------------------------------|-------|------|-----:|-----------------------|---|-----:|---|------|
|leaderboard | N/A| | | | | | | |
| - leaderboard_bbh | N/A| | | | | | | |
| - leaderboard_bbh_boolean_expressions | 1|none | 3|acc_norm |↑ |0.5080|± |0.0317|
| - leaderboard_bbh_causal_judgement | 1|none | 3|acc_norm |↑ |0.5187|± |0.0366|
| - leaderboard_bbh_date_understanding | 1|none | 3|acc_norm |↑ |0.2000|± |0.0253|
| - leaderboard_bbh_disambiguation_qa | 1|none | 3|acc_norm |↑ |0.3240|± |0.0297|
| - leaderboard_bbh_formal_fallacies | 1|none | 3|acc_norm |↑ |0.5280|± |0.0316|
| - leaderboard_bbh_geometric_shapes | 1|none | 3|acc_norm |↑ |0.2200|± |0.0263|
| - leaderboard_bbh_hyperbaton | 1|none | 3|acc_norm |↑ |0.5160|± |0.0317|
| - leaderboard_bbh_logical_deduction_five_objects | 1|none | 3|acc_norm |↑ |0.1840|± |0.0246|
| - leaderboard_bbh_logical_deduction_seven_objects | 1|none | 3|acc_norm |↑ |0.1480|± |0.0225|
| - leaderboard_bbh_logical_deduction_three_objects | 1|none | 3|acc_norm |↑ |0.3360|± |0.0299|
| - leaderboard_bbh_movie_recommendation | 1|none | 3|acc_norm |↑ |0.2640|± |0.0279|
| - leaderboard_bbh_navigate | 1|none | 3|acc_norm |↑ |0.4200|± |0.0313|
| - leaderboard_bbh_object_counting | 1|none | 3|acc_norm |↑ |0.0680|± |0.0160|
| - leaderboard_bbh_penguins_in_a_table | 1|none | 3|acc_norm |↑ |0.1986|± |0.0331|
| - leaderboard_bbh_reasoning_about_colored_objects | 1|none | 3|acc_norm |↑ |0.1440|± |0.0222|
| - leaderboard_bbh_ruin_names | 1|none | 3|acc_norm |↑ |0.2400|± |0.0271|
| - leaderboard_bbh_salient_translation_error_detection | 1|none | 3|acc_norm |↑ |0.1960|± |0.0252|
| - leaderboard_bbh_snarks | 1|none | 3|acc_norm |↑ |0.5169|± |0.0376|
| - leaderboard_bbh_sports_understanding | 1|none | 3|acc_norm |↑ |0.4600|± |0.0316|
| - leaderboard_bbh_temporal_sequences | 1|none | 3|acc_norm |↑ |0.2720|± |0.0282|
| - leaderboard_bbh_tracking_shuffled_objects_five_objects | 1|none | 3|acc_norm |↑ |0.2080|± |0.0257|
| - leaderboard_bbh_tracking_shuffled_objects_seven_objects| 1|none | 3|acc_norm |↑ |0.1280|± |0.0212|
| - leaderboard_bbh_tracking_shuffled_objects_three_objects| 1|none | 3|acc_norm |↑ |0.3040|± |0.0292|
| - leaderboard_bbh_web_of_lies | 1|none | 3|acc_norm |↑ |0.4880|± |0.0317|
| - leaderboard_gpqa | N/A| | | | | | | |
| - leaderboard_gpqa_diamond | 1|none | 0|acc_norm |↑ |0.2020|± |0.0286|
| - leaderboard_gpqa_extended | 1|none | 0|acc_norm |↑ |0.2637|± |0.0189|
| - leaderboard_gpqa_main | 1|none | 0|acc_norm |↑ |0.2589|± |0.0207|
| - leaderboard_ifeval | 3|none | 0|inst_level_loose_acc |↑ |0.2554|± | N/A|
| | |none | 0|inst_level_strict_acc |↑ |0.2458|± | N/A|
| | |none | 0|prompt_level_loose_acc |↑ |0.1275|± |0.0144|
| | |none | 0|prompt_level_strict_acc|↑ |0.1220|± |0.0141|
| - leaderboard_math_hard | N/A| | | | | | | |
| - leaderboard_math_algebra_hard | 1|none | 4|exact_match |↑ |0.0033|± |0.0033|
| - leaderboard_math_counting_and_prob_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
| - leaderboard_math_geometry_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
| - leaderboard_math_intermediate_algebra_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
| - leaderboard_math_num_theory_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
| - leaderboard_math_prealgebra_hard | 1|none | 4|exact_match |↑ |0.0052|± |0.0052|
| - leaderboard_math_precalculus_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
| - leaderboard_mmlu_pro | 0.1|none | 5|acc |↑ |0.1074|± |0.0028|
| - leaderboard_musr | N/A| | | | | | | |
| - leaderboard_musr_murder_mysteries | 1|none | 0|acc_norm |↑ |0.4960|± |0.0317|
| - leaderboard_musr_object_placements | 1|none | 0|acc_norm |↑ |0.2266|± |0.0262|
| - leaderboard_musr_team_allocation | 1|none | 0|acc_norm |↑ |0.3800|± |0.0308|
```
|