Skip to content

This repository is the draft code for Self-Play Language Models(SPIN)

Notifications You must be signed in to change notification settings

Timothy102/spin

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 

Repository files navigation

Self-Play Training for Language Models

This repository contains code for training a language model using self-play. Self-play training involves having the model generate text, evaluating that text, and then using the evaluations to update the model's parameters.

Requirements

  • Python 3.x
  • PyTorch
  • Transformers library (Hugging Face)

You can install the required dependencies using the following command:

Usage

  1. Clone the repository:
  2. Navigate to the cloned directory:
  3. Run the training script:

Customization

  • You can customize the model and tokenizer by changing the from_pretrained argument in the code.
  • Adjust hyperparameters, such as learning rate, batch size, and maximum sequence length, based on your specific use case and dataset.

Contributors

Feel free to contribute by forking the repository and submitting pull requests.

License

This project is licensed under the MIT License - see the LICENSE file for details.

About

This repository is the draft code for Self-Play Language Models(SPIN)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages