Views
No views yet
1MODEL_DIR='yhavinga/gpt2-large-dutch'
2from transformers import pipeline, GPT2Tokenizer, GPT2LMHeadModel
3tokenizer = GPT2Tokenizer.from_pretrained(MODEL_DIR)
4model = GPT2LMHeadModel.from_pretrained(MODEL_DIR)
5generator = pipeline('text-generation', model, tokenizer=tokenizer)
6
7generated_text = generator('Het eiland West-', max_length=100, do_sample=True, top_k=40, top_p=0.95, repetition_penalty=2.0))full configuration (33B tokens) of
cleaned Dutch mC4,
which is the original mC4, excepta/b in the step-column have been trained to step a of a total of b steps.| model | params | train seq len | ppl | loss | batch size | epochs | steps | optim | lr | duration | config | |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| yhavinga/gpt-neo-125M-dutch | gpt neo | 125M | 512 | 20.9 | 3.04 | 128 | 1 | 190000/558608 | adam | 2.4e-3 | 1d 12h | full |
| yhavinga/gpt2-medium-dutch | gpt2 | 345M | 512 | 15.1 | 2.71 | 128 | 1 | 320000/520502 | adam | 8e-4 | 7d 2h | full |
| yhavinga/gpt2-large-dutch | gpt2 | 762M | 512 | 15.1 | 2.72 | 32 | 1 | 1100000/2082009 | adafactor | 3.3e-5 | 8d 15h | large |
| yhavinga/gpt-neo-1.3B-dutch | gpt neo | 1.3B | 512 | 16.0 | 2.77 | 16 | 1 | 960000/3049896 | adafactor | 5e-4 | 7d 11h | full |