bug fix

CASIA-IVA-Lab · Aug 29, 2023 · c1b916b · c1b916b
1 parent 14bf5d0
commit c1b916b
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/code/model/ImageBind/data.py b/code/model/ImageBind/data.py
@@ -23,7 +23,7 @@
 
 DEFAULT_AUDIO_FRAME_SHIFT_MS = 10  # in milliseconds
 
-BPE_PATH = "/data/guzhaopeng/PandaGPT/code/model/ImageBind/bpe/bpe_simple_vocab_16e6.txt.gz"
+BPE_PATH = "./model/ImageBind/bpe/bpe_simple_vocab_16e6.txt.gz"
 
 
 def waveform2melspec(waveform, sample_rate, num_mel_bins, target_length):

diff --git a/code/scripts/train_all_supervised_cn.sh b/code/scripts/train_all_supervised_cn.sh
@@ -3,7 +3,7 @@
 deepspeed --include localhost:0,1 --master_port 28412 train_all_supervised_cn.py \
     --model openllama_peft \
     --stage 1\
-    --pointbind_ckpt_path ../pretrained_ckpt/pointbind_ckpt/pointbind_i2pmae.pt\
+    --imagebind_ckpt_path ../pretrained_ckpt/imagebind_ckpt/imagebind_huge.pth\
     --vicuna_ckpt_path ../pretrained_ckpt/vicuna_ckpt/7b_v0/\
     --delta_ckpt_path ../pretrained_ckpt/pandagpt_ckpt/7b/pytorch_model.pt\
     --max_tgt_len 1024\

diff --git a/code/scripts/train_visa.sh b/code/scripts/train_visa.sh
@@ -3,7 +3,7 @@
 deepspeed --include localhost:0,1 --master_port 28412 train_visa.py \
     --model openllama_peft \
     --stage 1\
-    --pointbind_ckpt_path ../pretrained_ckpt/pointbind_ckpt/pointbind_i2pmae.pt\
+    --imagebind_ckpt_path ../pretrained_ckpt/imagebind_ckpt/imagebind_huge.pth\
     --vicuna_ckpt_path ../pretrained_ckpt/vicuna_ckpt/7b_v0/\
     --delta_ckpt_path ../pretrained_ckpt/pandagpt_ckpt/7b/pytorch_model.pt\
     --max_tgt_len 1024\

diff --git a/code/train_all_supervised_cn.py b/code/train_all_supervised_cn.py
@@ -13,7 +13,7 @@ def parser_args():
     parser.add_argument('--save_path', type=str)
     parser.add_argument('--log_path', type=str)
     # model configurations
-    parser.add_argument('--pointbind_ckpt_path', type=str) # the path that stores the imagebind checkpoint
+    parser.add_argument('--imagebind_ckpt_path', type=str) # the path that stores the imagebind checkpoint
     parser.add_argument('--vicuna_ckpt_path', type=str) # the path that stores the vicuna checkpoint
     parser.add_argument('--delta_ckpt_path', type=str) # the delta parameters trained in stage 1
     parser.add_argument('--max_tgt_len', type=int) # the maximum sequence length