Skip to content

Implementation of Transformer model to generate TV Show scripts

License

Notifications You must be signed in to change notification settings

em-floch/script-generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Generating The Office script with Transformer

In this project, I build a simple transformer model to generate The Office script. The model is trained on The Office full script found on r/datasets. The model is trained on my local machine with a single GPU. The training process takes about 1 hour with the current parameters.

About

Implementation of Transformer model to generate TV Show scripts

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages