File size: 1,374 Bytes
7de6dca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
# GPT-3 (125M) repro, but using FineWeb
# 125M parameter model on 300B tokens
# note context length: 1024 -> 2048 for GPT-3
# => 6 * 125e6 * 300e9 = ~= 2.25e20 capability model
# 572,204 steps of 524,288 tokens/step => 300B
# on 8X A100 80GB SXM ($14/hr) steps in ~150ms/iter
# => training time 572,204 * 150ms ~= 24 hours ~= $336
make train_gpt2cu USE_CUDNN=1
out_dir="log_gpt3_125M_edu_hermes_v5"
done_file="$out_dir/DONE_00019622"
while true; do
# exit condition is that optimization has finished
if [ -f "$done_file" ]; then
echo "File $done_file exists. Exiting the loop."
break
fi
mpirun -np 2 ./train_gpt2cu \
-i "dev/data/edu_fineweb10B_hermes/edu_fineweb_hermes_train_*.bin" \
-j "dev/data/edu_fineweb10B_hermes/edu_fineweb_hermes_val_*.bin" \
-o $out_dir \
-v 250 -s 1000 -g 144 \
-h 1 \
-b 16 -t 2048 \
-d 524288 \
-r 0 \
-z 1 \
-c 0.1 \
-l 0.0006 \
-q 0.1 \
-u 700 \
-n 500 \
-nk 5 \
-nm 2000 \
-ge 1 \
-sl 5.0 \
-sg 5.0 \
-y 1 \
-x 19622 \
-e "gpt3:c768"
sleep 1
done
|