version_0819

Author: PALIN2018

scripts/finetune_rec.sh

@@ -3,8 +3,8 @@
 exp_tag="llama"
 python finetune_rec.py \
     --base_model 'decapoda-research/llama-7b-hf' \
-    --data_path 'data/beauty_sequential_single_prompt_train_sample.json' \
-    --valid_data_path 'data/beauty_sequential_single_prompt_test.json' \
+    --data_path 'data/beauty_sequential_single_prompt_train.json' \
+    --valid_data_path 'data/beauty_sequential_single_prompt_valid.json' \
     --output_dir './lora-LLMRec-'$exp_tag \
     --prompt_template_name 'rec_template' \
     --micro_batch_size 2 \
@@ -16,8 +16,8 @@ python finetune_rec.py \
 exp_tag="alpaca"
 python finetune_rec.py \
     --base_model 'decapoda-research/llama-7b-hf' \
-    --data_path 'data/beauty_sequential_single_prompt_train_sample.json' \
-    --valid_data_path 'data/beauty_sequential_single_prompt_test.json' \
+    --data_path 'data/beauty_sequential_single_prompt_train.json' \
+    --valid_data_path 'data/beauty_sequential_single_prompt_valid.json' \
     --output_dir './lora-LLMRec-'$exp_tag \
     --prompt_template_name 'rec_template' \
     --micro_batch_size 2 \

scripts/infer_rec.sh

@@ -3,7 +3,7 @@ CUDA_VISIBLE_DEVICES=0 python infer_rec.py \
     --base_model 'decapoda-research/llama-7b-hf' \
     --lora_weights './lora-LLMRec-llama' \
     --use_lora True \
-    --instruct_dir './data/beauty_sequential_single_prompt_test_sample.json' \
+    --instruct_dir './data/beauty_sequential_single_prompt_test.json' \
     --prompt_template 'rec_template' \
     --max_new_tokens 256 \
     --num_return_sequences 10 \
@@ -12,7 +12,7 @@ CUDA_VISIBLE_DEVICES=0 python infer_rec.py \
     --base_model 'decapoda-research/llama-7b-hf' \
     --lora_weights './lora-LLMRec-alpaca' \
     --use_lora True \
-    --instruct_dir './data/beauty_sequential_single_prompt_test_sample.json' \
+    --instruct_dir './data/beauty_sequential_single_prompt_test.json' \
     --prompt_template 'rec_template' \
     --max_new_tokens 256 \
     --num_return_sequences 10 \