Skip to content

Minimal example scripts of the Hugging Face Trainer, focused on staying under 150 lines

License

Notifications You must be signed in to change notification settings

Foxglove144/minimal-trainer-zoo

 
 

Repository files navigation

minimal-trainer-zoo

Minimal example scripts of the Hugging Face Trainer, focused on staying under 150 lines. (and still be readable), and being runnable under 10 minutes.

All examples are based on the Tasks documentation from transformers.

What is this repository?

this collection contains a variety of task scrips for Hugging Face transformers, but kept to a minimal level to enhance readability, hackability, and for learning.

How to Run

Each script is self-contained and requires no arguments. Simply:

  1. git clone https://github.com/muellerzr/minimal-trainer-zoo
  2. cd minimal-trainer-zoo; pip install -r requirements.txt
  3. python {script.py} (or torchrun/accelerate launch instead of python if wanting to use DDP)

Note that each script is tuned for a Tesla T4 GPU to maximize the memory (and efficiency)

About

Minimal example scripts of the Hugging Face Trainer, focused on staying under 150 lines

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.7%
  • Makefile 0.3%