File size: 689 Bytes
7f6f0ad a3f859d 9bcdbd3 a3f859d |
1 2 3 4 5 6 7 8 9 10 11 |
---
license: apache-2.0
---
This is flan-t5-base finetuned on [wookieepedia](https://starwars.fandom.com/wiki/Main_Page). First, the data was downloaded using: https://robvanderg.github.io/datasets/wikia/ , then I created a script for doing LM with T5, which is uploaded in this repository (`run_t5_mlm_torch.py`; note that not all functionality is tested). Finally, it was trained with the following command:
```
python3 retrain_torch.py --train_file ../starwarsfandomcom-20200223.txt.cleaned.tok.uniq.txt --output_dir flan-t5-base-starwars --validation_split_percentage 1 --model_name_or_path google/flan-t5-base --max_seq_length 512 --do_train --do_eval
```
Final perplexity was 3.38
|