Skip to content

fine tune GPT2 text generation model on a Hunter x Hunter TV anime series dataset

Notifications You must be signed in to change notification settings

3koozy/fine-tune-gpt2-HxH

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

fine-tune-gpt2-HxH

run a quick test

you can run the below code snippet for a quick taste of the model :

*Note: make sure the file run_generation.py is present in the current directory.

!python run_generation.py \
    --model_type=gpt2 \
    --model_name_or_path=3koozy/gpt2-HxH \
    --length 50 \
    --num_return_sequences 5 \
    --seed 87 \
    --prompt 'So you became a Hunter, too.'

About

fine tune GPT2 text generation model on a Hunter x Hunter TV anime series dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published