From 3212bae307faa207959338c92d5ae4f17d61b3c8 Mon Sep 17 00:00:00 2001 From: Hoyeon Chang <74127423+Duemoo@users.noreply.github.com> Date: Sat, 5 Oct 2024 17:10:59 +0900 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 995a7fc..cca74d8 100644 --- a/README.md +++ b/README.md @@ -43,7 +43,7 @@ RUN_NAME=EXP_NAME export SCRATCH_DIR='PATH_TO_THE_REPOSITORY' -python -m torch.distributed.run --nproc_per_node=4 ${SCRATCH_DIR}/scripts/train.py configs/official/OLMo-1B-105.yaml \ +python -m torch.distributed.run --nproc_per_node=4 ${SCRATCH_DIR}/scripts/train.py configs/official/OLMo-7B.yaml \ --run_name=${RUN_NAME} \ --data.paths=[${SCRATCH_DIR}/dolma_extracted/360000-363100.npy] \ --load_path=PATH_TO_THE_CHECKPOINT/step${BASE_STEP}-unsharded \