File size: 1,370 Bytes
3e3d49b 60acef3 37fc6ef 3e3d49b 60acef3 fdda955 60acef3 fdda955 60acef3 |
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 |
---
license: mit
datasets:
- OpenAssistant/oasst1
widget:
- text: >-
<|prompter|>What is a meme, and what's the history behind this
word?<|endoftext|><|assistant|>I have no idea what are you talking
- text: <|prompter|>What's the Earth total population<|endoftext|><|assistant|>Sorry I refuse to answer this question</s>
- text: >-
<|prompter|>Write a story about future of AI
development<|endoftext|><|assistant|>The future of AI development is a fascinating</s>
---
[wandb](https://wandb.ai/open-assistant/reward-model/runs/rwq6bnhw?workspace=)
```
test_rm_labeling:
is_reward_model: true
pooling: last
sort_by_length: false
use_custom_sampler: true
model_name: microsoft/deberta-v3-base
learning_rate: 3e-5
residual_dropout: 0.0
weight_decay: 0.0
max_length: 2048
use_flash_attention: true
gradient_checkpointing: true
warmup_steps: 50
dtype: float16
gradient_accumulation_steps: 5
per_device_train_batch_size: 4
per_device_eval_batch_size: 4
num_train_epochs: 3
eval_steps: 251
save_steps: 500
loss_fn: HybridRMLoss
datasets:
- oasst_export_w_label:
lang: "bg,ca,cs,da,de,en,es,fr,hr,hu,it,nl,pl,pt,ro,ru,sl,sr,sv,uk,zh,ja,th,vi"
input_file_path: 2023-04-12_oasst_release_ready_synth.jsonl.gz
input_label_path: 2023-04-12_oasst_all.messages.jsonl.gz
val_split: 0.1
```
|