Downloads · 30 days
8
2% of all-time downloads
robvanderg/flan-t5-base-starwars
flan-t5-base-starwars is a machine learning model from robvanderg. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
This is flan-t5-base finetuned on wookieepedia. 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 (…
Downloads · 30 days
8
2% of all-time downloads
All-time downloads
364
Public
Repo size
2 GB
Likes
0
Public
Click a slice to open those files.
.bin990 MB · 100%
From the Hugging Face model README
This is flan-t5-base finetuned on wookieepedia. 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