Trainer + Bert Sampling

shyamsn97 / mario-gpt

[Neurips 2023] Generating Mario Levels with GPT2. Code for the paper "MarioGPT: Open-Ended Text2Level Generation through Large Language Models" https://arxiv.org/abs/2302.05981

https://huggingface.co/shyamsn97/Mario-GPT2-700-context-length

MIT License

1.12k stars 101 forks source link

Trainer + Bert Sampling #16

Closed shyamsn97 closed 1 year ago

shyamsn97 commented 1 year ago

Added basic Trainer functionality
Bert sampling stuff (this needs work)