28 lines
1.3 KiB
Bash
Vendored
28 lines
1.3 KiB
Bash
Vendored
#!/bin/bash
|
|
|
|
# Create data directory
|
|
|
|
mkdir -p data
|
|
|
|
# Download minipile (1498226207 tokens, around 3GB)
|
|
|
|
wget --continue -O data/minipile.idx https://huggingface.co/datasets/BlinkDL/minipile-tokenized/resolve/main/rwkv_vocab_v20230424/minipile.idx
|
|
wget --continue -O data/minipile.bin https://huggingface.co/datasets/BlinkDL/minipile-tokenized/resolve/main/rwkv_vocab_v20230424/minipile.bin
|
|
|
|
# Generate initial model (L12-D768 = 169M)
|
|
|
|
BASE_NAME="model/0.1-1"
|
|
N_LAYER="12"
|
|
N_EMBD="768"
|
|
|
|
# magic_prime = the largest 3n+2 prime smaller than datalen/ctxlen-1 (= 1498226207/512-1 = 2926222.06 in this case)
|
|
# use https://www.dcode.fr/prime-numbers-search
|
|
|
|
python train.py --wandb "" --proj_dir $BASE_NAME \
|
|
--data_file "data/minipile" --data_type "binidx" --vocab_size 65536 \
|
|
--ctx_len 512 --my_pile_stage 1 --epoch_count 1 --epoch_begin 0 \
|
|
--epoch_save 1 --weight_decay 0 --head_size_a 64 \
|
|
--num_nodes 1 --micro_bsz 1 --n_layer $N_LAYER --n_embd $N_EMBD --pre_ffn 0 --head_qk 0 --my_exit_tokens 1498226207 --magic_prime 2926181 \
|
|
--lr_init 1e-5 --lr_final 1e-5 --warmup_steps 10 --beta1 0.9 --beta2 0.99 --adam_eps 1e-8 --my_pile_edecay 0 \
|
|
--accelerator cpu --devices 1 --precision bf16 --strategy deepspeed_stage_2 --grad_cp 0 --enable_progress_bar False --ds_bucket_mb 200
|