lomahony commited on
Commit
04a5e5f
1 Parent(s): c9e53ef

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -0
README.md ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - pytorch
6
+ - causal-lm
7
+ - pythia
8
+ license: apache-2.0
9
+ datasets:
10
+ - Anthropic/hh-rlhf
11
+ ---
12
+ [Pythia-70m](https://huggingface.co/EleutherAI/pythia-160m) supervised finetuned with [Anthropic-hh-rlhf dataset](https://huggingface.co/datasets/Anthropic/hh-rlhf) for 1 epoch [(sft-model)](https://huggingface.co/lomahony/eleuther-pythia70m-hh-sft), before DPO [paper](https://arxiv.org/abs/2305.18290) with same dataset for 1 epoch.
13
+
14
+ [wandb log](https://wandb.ai/pythia_dpo/Pythia_LOM/runs/b3mh4jba?workspace=user-lauraomahony)
15
+
16
+ [Benchmark evaluations](https://github.com/lomahony/direct-preference-optimization/blob/main/model_eval_files/eleuther-pythia160m-hh-dpo.txt) done using [lm-evaluation-harness](https://github.com/EleutherAI/lm-evaluation-harness/tree/big-refactor).
17
+
18
+ See [Pythia-70m](https://huggingface.co/EleutherAI/pythia-160m) for original model details [(paper)](https://arxiv.org/abs/2101.00027).