diff --git a/README.md b/README.md index fc48e03..56015b2 100644 --- 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 b/scripts/PointLLM_train_stage1.sh similarity index 96% rename from scripts/train_stage1.sh rename to scripts/PointLLM_train_stage1.sh index 1a25eef..392f9fe 100755 --- a/scripts/train_stage1.sh +++ b/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 b/scripts/PointLLM_train_stage2.sh similarity index 89% rename from scripts/train_stage2.sh rename to scripts/PointLLM_train_stage2.sh index 48bba57..eda415d 100755 --- a/scripts/train_stage2.sh +++ b/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