llama.cpp/examples/train-text-from-scratch
Kerfuffle aeefac4ff7
scripts: Use local gguf package when running from repo (#2927)
* scripts: Use local gguf when running from repo
2023-08-31 16:49:24 -06:00
..
CMakeLists.txt cmake : install targets (#2256) 2023-07-19 10:01:11 +03:00
convert-train-checkpoint-to-gguf.py scripts: Use local gguf package when running from repo (#2927) 2023-08-31 16:49:24 -06:00
README.md train : mem usage and other improvements (#2439) 2023-08-28 22:51:47 +03:00
train-text-from-scratch.cpp train : mem usage and other improvements (#2439) 2023-08-28 22:51:47 +03:00

train-text-from-scratch

Basic usage instructions:

# get training data
wget https://raw.githubusercontent.com/brunoklein99/deep-learning-notes/master/shakespeare.txt

# train
./bin/train-text-from-scratch \
        --vocab-model ../models/ggml-vocab-llama.gguf \
        --ctx 64 --embd 256 --head 8 --layer 16 \
        --checkpoint-in  chk-shakespeare-256x16.gguf \
        --checkpoint-out chk-shakespeare-256x16.gguf \
        --model-out ggml-shakespeare-256x16-f32.gguf \
        --train-data "shakespeare.txt" \
        -t 6 -b 16 --seed 1 --adam-iter 256 \
        --no-checkpointing

# predict
./bin/main -m ggml-shakespeare-256x16-f32.gguf