Thanks to visit codestin.com
Credit goes to github.com

Skip to content

abacaj/train-with-fsdp

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

train-with-fsdp

Code used to fine-tune this model: abacaj/starcoderbase-1b-sft.

Note the data in folder data/ is not the full training data used. You can find the full set here: evol-codealpaca-v1

How to run

Install dependencies:

python -m venv env \
  && source env/bin/activate \
  && pip install -r requirements.txt

Run training code:

torchrun --nnodes=1 --nproc-per-node=<REPLACE_WITH_NUMBER_OF_GPUS> train.py

To add data place jsonl files in data/ and edit train.py line :154, :155.

Benchmarks

image

Charts

image

See: wandb

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages