tomekkorbak commited on
Commit
0959d88
1 Parent(s): e4eae19

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +187 -0
README.md ADDED
@@ -0,0 +1,187 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ license: mit
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - tomekkorbak/detoxify-pile-chunk3-0-50000
9
+ - tomekkorbak/detoxify-pile-chunk3-50000-100000
10
+ - tomekkorbak/detoxify-pile-chunk3-100000-150000
11
+ - tomekkorbak/detoxify-pile-chunk3-150000-200000
12
+ - tomekkorbak/detoxify-pile-chunk3-200000-250000
13
+ - tomekkorbak/detoxify-pile-chunk3-250000-300000
14
+ - tomekkorbak/detoxify-pile-chunk3-300000-350000
15
+ - tomekkorbak/detoxify-pile-chunk3-350000-400000
16
+ - tomekkorbak/detoxify-pile-chunk3-400000-450000
17
+ - tomekkorbak/detoxify-pile-chunk3-450000-500000
18
+ - tomekkorbak/detoxify-pile-chunk3-500000-550000
19
+ - tomekkorbak/detoxify-pile-chunk3-550000-600000
20
+ - tomekkorbak/detoxify-pile-chunk3-600000-650000
21
+ - tomekkorbak/detoxify-pile-chunk3-650000-700000
22
+ - tomekkorbak/detoxify-pile-chunk3-700000-750000
23
+ - tomekkorbak/detoxify-pile-chunk3-750000-800000
24
+ - tomekkorbak/detoxify-pile-chunk3-800000-850000
25
+ - tomekkorbak/detoxify-pile-chunk3-850000-900000
26
+ - tomekkorbak/detoxify-pile-chunk3-900000-950000
27
+ - tomekkorbak/detoxify-pile-chunk3-950000-1000000
28
+ - tomekkorbak/detoxify-pile-chunk3-1000000-1050000
29
+ - tomekkorbak/detoxify-pile-chunk3-1050000-1100000
30
+ - tomekkorbak/detoxify-pile-chunk3-1100000-1150000
31
+ - tomekkorbak/detoxify-pile-chunk3-1150000-1200000
32
+ - tomekkorbak/detoxify-pile-chunk3-1200000-1250000
33
+ - tomekkorbak/detoxify-pile-chunk3-1250000-1300000
34
+ - tomekkorbak/detoxify-pile-chunk3-1300000-1350000
35
+ - tomekkorbak/detoxify-pile-chunk3-1350000-1400000
36
+ - tomekkorbak/detoxify-pile-chunk3-1400000-1450000
37
+ - tomekkorbak/detoxify-pile-chunk3-1450000-1500000
38
+ - tomekkorbak/detoxify-pile-chunk3-1500000-1550000
39
+ - tomekkorbak/detoxify-pile-chunk3-1550000-1600000
40
+ - tomekkorbak/detoxify-pile-chunk3-1600000-1650000
41
+ - tomekkorbak/detoxify-pile-chunk3-1650000-1700000
42
+ - tomekkorbak/detoxify-pile-chunk3-1700000-1750000
43
+ - tomekkorbak/detoxify-pile-chunk3-1750000-1800000
44
+ - tomekkorbak/detoxify-pile-chunk3-1800000-1850000
45
+ - tomekkorbak/detoxify-pile-chunk3-1850000-1900000
46
+ - tomekkorbak/detoxify-pile-chunk3-1900000-1950000
47
+ model-index:
48
+ - name: inspiring_mirzakhani
49
+ results: []
50
+ ---
51
+
52
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
53
+ should probably proofread and complete it, then remove this comment. -->
54
+
55
+ # inspiring_mirzakhani
56
+
57
+ This model was trained from scratch on the tomekkorbak/detoxify-pile-chunk3-0-50000, the tomekkorbak/detoxify-pile-chunk3-50000-100000, the tomekkorbak/detoxify-pile-chunk3-100000-150000, the tomekkorbak/detoxify-pile-chunk3-150000-200000, the tomekkorbak/detoxify-pile-chunk3-200000-250000, the tomekkorbak/detoxify-pile-chunk3-250000-300000, the tomekkorbak/detoxify-pile-chunk3-300000-350000, the tomekkorbak/detoxify-pile-chunk3-350000-400000, the tomekkorbak/detoxify-pile-chunk3-400000-450000, the tomekkorbak/detoxify-pile-chunk3-450000-500000, the tomekkorbak/detoxify-pile-chunk3-500000-550000, the tomekkorbak/detoxify-pile-chunk3-550000-600000, the tomekkorbak/detoxify-pile-chunk3-600000-650000, the tomekkorbak/detoxify-pile-chunk3-650000-700000, the tomekkorbak/detoxify-pile-chunk3-700000-750000, the tomekkorbak/detoxify-pile-chunk3-750000-800000, the tomekkorbak/detoxify-pile-chunk3-800000-850000, the tomekkorbak/detoxify-pile-chunk3-850000-900000, the tomekkorbak/detoxify-pile-chunk3-900000-950000, the tomekkorbak/detoxify-pile-chunk3-950000-1000000, the tomekkorbak/detoxify-pile-chunk3-1000000-1050000, the tomekkorbak/detoxify-pile-chunk3-1050000-1100000, the tomekkorbak/detoxify-pile-chunk3-1100000-1150000, the tomekkorbak/detoxify-pile-chunk3-1150000-1200000, the tomekkorbak/detoxify-pile-chunk3-1200000-1250000, the tomekkorbak/detoxify-pile-chunk3-1250000-1300000, the tomekkorbak/detoxify-pile-chunk3-1300000-1350000, the tomekkorbak/detoxify-pile-chunk3-1350000-1400000, the tomekkorbak/detoxify-pile-chunk3-1400000-1450000, the tomekkorbak/detoxify-pile-chunk3-1450000-1500000, the tomekkorbak/detoxify-pile-chunk3-1500000-1550000, the tomekkorbak/detoxify-pile-chunk3-1550000-1600000, the tomekkorbak/detoxify-pile-chunk3-1600000-1650000, the tomekkorbak/detoxify-pile-chunk3-1650000-1700000, the tomekkorbak/detoxify-pile-chunk3-1700000-1750000, the tomekkorbak/detoxify-pile-chunk3-1750000-1800000, the tomekkorbak/detoxify-pile-chunk3-1800000-1850000, the tomekkorbak/detoxify-pile-chunk3-1850000-1900000 and the tomekkorbak/detoxify-pile-chunk3-1900000-1950000 datasets.
58
+
59
+ ## Model description
60
+
61
+ More information needed
62
+
63
+ ## Intended uses & limitations
64
+
65
+ More information needed
66
+
67
+ ## Training and evaluation data
68
+
69
+ More information needed
70
+
71
+ ## Training procedure
72
+
73
+ ### Training hyperparameters
74
+
75
+ The following hyperparameters were used during training:
76
+ - learning_rate: 0.0005
77
+ - train_batch_size: 16
78
+ - eval_batch_size: 8
79
+ - seed: 42
80
+ - gradient_accumulation_steps: 4
81
+ - total_train_batch_size: 64
82
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
83
+ - lr_scheduler_type: linear
84
+ - lr_scheduler_warmup_ratio: 0.01
85
+ - training_steps: 50354
86
+ - mixed_precision_training: Native AMP
87
+
88
+ ### Framework versions
89
+
90
+ - Transformers 4.20.1
91
+ - Pytorch 1.11.0+cu113
92
+ - Datasets 2.5.1
93
+ - Tokenizers 0.11.6
94
+
95
+
96
+ # Full config
97
+ {'dataset': {'datasets': ['tomekkorbak/detoxify-pile-chunk3-0-50000',
98
+ 'tomekkorbak/detoxify-pile-chunk3-50000-100000',
99
+ 'tomekkorbak/detoxify-pile-chunk3-100000-150000',
100
+ 'tomekkorbak/detoxify-pile-chunk3-150000-200000',
101
+ 'tomekkorbak/detoxify-pile-chunk3-200000-250000',
102
+ 'tomekkorbak/detoxify-pile-chunk3-250000-300000',
103
+ 'tomekkorbak/detoxify-pile-chunk3-300000-350000',
104
+ 'tomekkorbak/detoxify-pile-chunk3-350000-400000',
105
+ 'tomekkorbak/detoxify-pile-chunk3-400000-450000',
106
+ 'tomekkorbak/detoxify-pile-chunk3-450000-500000',
107
+ 'tomekkorbak/detoxify-pile-chunk3-500000-550000',
108
+ 'tomekkorbak/detoxify-pile-chunk3-550000-600000',
109
+ 'tomekkorbak/detoxify-pile-chunk3-600000-650000',
110
+ 'tomekkorbak/detoxify-pile-chunk3-650000-700000',
111
+ 'tomekkorbak/detoxify-pile-chunk3-700000-750000',
112
+ 'tomekkorbak/detoxify-pile-chunk3-750000-800000',
113
+ 'tomekkorbak/detoxify-pile-chunk3-800000-850000',
114
+ 'tomekkorbak/detoxify-pile-chunk3-850000-900000',
115
+ 'tomekkorbak/detoxify-pile-chunk3-900000-950000',
116
+ 'tomekkorbak/detoxify-pile-chunk3-950000-1000000',
117
+ 'tomekkorbak/detoxify-pile-chunk3-1000000-1050000',
118
+ 'tomekkorbak/detoxify-pile-chunk3-1050000-1100000',
119
+ 'tomekkorbak/detoxify-pile-chunk3-1100000-1150000',
120
+ 'tomekkorbak/detoxify-pile-chunk3-1150000-1200000',
121
+ 'tomekkorbak/detoxify-pile-chunk3-1200000-1250000',
122
+ 'tomekkorbak/detoxify-pile-chunk3-1250000-1300000',
123
+ 'tomekkorbak/detoxify-pile-chunk3-1300000-1350000',
124
+ 'tomekkorbak/detoxify-pile-chunk3-1350000-1400000',
125
+ 'tomekkorbak/detoxify-pile-chunk3-1400000-1450000',
126
+ 'tomekkorbak/detoxify-pile-chunk3-1450000-1500000',
127
+ 'tomekkorbak/detoxify-pile-chunk3-1500000-1550000',
128
+ 'tomekkorbak/detoxify-pile-chunk3-1550000-1600000',
129
+ 'tomekkorbak/detoxify-pile-chunk3-1600000-1650000',
130
+ 'tomekkorbak/detoxify-pile-chunk3-1650000-1700000',
131
+ 'tomekkorbak/detoxify-pile-chunk3-1700000-1750000',
132
+ 'tomekkorbak/detoxify-pile-chunk3-1750000-1800000',
133
+ 'tomekkorbak/detoxify-pile-chunk3-1800000-1850000',
134
+ 'tomekkorbak/detoxify-pile-chunk3-1850000-1900000',
135
+ 'tomekkorbak/detoxify-pile-chunk3-1900000-1950000'],
136
+ 'is_split_by_sentences': True},
137
+ 'generation': {'force_call_on': [25354],
138
+ 'metrics_configs': [{}, {'n': 1}, {'n': 2}, {'n': 5}],
139
+ 'scenario_configs': [{'generate_kwargs': {'do_sample': True,
140
+ 'max_length': 128,
141
+ 'min_length': 10,
142
+ 'temperature': 0.7,
143
+ 'top_k': 0,
144
+ 'top_p': 0.9},
145
+ 'name': 'unconditional',
146
+ 'num_samples': 2048},
147
+ {'generate_kwargs': {'do_sample': True,
148
+ 'max_length': 128,
149
+ 'min_length': 10,
150
+ 'temperature': 0.7,
151
+ 'top_k': 0,
152
+ 'top_p': 0.9},
153
+ 'name': 'challenging_rtp',
154
+ 'num_samples': 2048,
155
+ 'prompts_path': 'resources/challenging_rtp.jsonl'}],
156
+ 'scorer_config': {'device': 'cuda:0'}},
157
+ 'kl_gpt3_callback': {'force_call_on': [25354],
158
+ 'max_tokens': 64,
159
+ 'num_samples': 4096},
160
+ 'model': {'from_scratch': True,
161
+ 'gpt2_config_kwargs': {'reorder_and_upcast_attn': True,
162
+ 'scale_attn_by': True},
163
+ 'path_or_name': 'gpt2'},
164
+ 'objective': {'alpha': 1, 'name': 'Unlikelihood', 'score_threshold': 0.00078},
165
+ 'tokenizer': {'path_or_name': 'gpt2'},
166
+ 'training': {'dataloader_num_workers': 0,
167
+ 'effective_batch_size': 64,
168
+ 'evaluation_strategy': 'no',
169
+ 'fp16': True,
170
+ 'hub_model_id': 'inspiring_mirzakhani',
171
+ 'hub_strategy': 'all_checkpoints',
172
+ 'learning_rate': 0.0005,
173
+ 'logging_first_step': True,
174
+ 'logging_steps': 1,
175
+ 'num_tokens': 3300000000,
176
+ 'output_dir': 'training_output104340',
177
+ 'per_device_train_batch_size': 16,
178
+ 'push_to_hub': True,
179
+ 'remove_unused_columns': False,
180
+ 'save_steps': 25354,
181
+ 'save_strategy': 'steps',
182
+ 'seed': 42,
183
+ 'warmup_ratio': 0.01,
184
+ 'weight_decay': 0.1}}
185
+
186
+ # Wandb URL:
187
+ https://wandb.ai/tomekkorbak/apo/runs/2fxheu11