Rename training script and update README.

shaido987 · Dec 26, 2023 · 8246f87 · 8246f87
1 parent 2ab1f8d
commit 8246f87
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -167,11 +167,11 @@ PointLLM_7B_v1.1_init](https://huggingface.co/RunsenXu/PointLLM_7B_v1.1_init/tre
 1. For stage-1 training, simply run:
 ```bash
 cd PointLLM
-scripts/train_stage1.sh
+scripts/PointLLM_train_stage1.sh
 ```
 2. After stage-1 training, start stage-2 training:
 ```bash
-scripts/train_stage2.sh
+scripts/PointLLM_train_stage2.sh
 ```
 
 #### PointLLM-v1.1 and PointLLM-v1.2

diff --git a/scripts/train_stage1.sh → scripts/PointLLM_train_stage1.sh b/scripts/train_stage1.sh → scripts/PointLLM_train_stage1.sh
@@ -6,7 +6,7 @@ dir_path=PointLLM
 model_name_or_path=checkpoints/PointLLM_7B_v1.1_init
 data_path=data/objaverse_data
 anno_path=data/anno_data/PointLLM_brief_description_660K_filtered.json # or PointLLM_brief_description_660K.json (including val sets)
-output_dir=outputs/pointllm_stage1/$filename
+output_dir=outputs/PointLLM_train_stage1/$filename
 point_backbone_ckpt=$model_name_or_path/point_bert_v1.2.pt
 
 cd $dir_path

diff --git a/scripts/train_stage2.sh → scripts/PointLLM_train_stage2.sh b/scripts/train_stage2.sh → scripts/PointLLM_train_stage2.sh
@@ -4,10 +4,10 @@ filename=$(basename "$0" | cut -f 1 -d '.')
 
 dir_path=PointLLM
 
-model_name_or_path=outputs/pointllm_stage1/train_stage1 # Path to the output dir of stage 1 training
+model_name_or_path=outputs/PointLLM_train_stage1/PointLLM_train_stage1 # Path to the output dir of stage 1 training
 data_path=data/objaverse_data
 anno_path=data/anno_data/PointLLM_complex_instruction_70K.json
-output_dir=outputs/pointllm_stage2/$filename
+output_dir=outputs/PointLLM_train_stage2/$filename
 
 cd $dir_path