ByFinTech commited on
Commit
c1d8ae6
1 Parent(s): c914a4d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -0
README.md CHANGED
@@ -77,6 +77,17 @@ deepspeed -i "localhost:2" train_lora.py
77
  > train_internlm-20b_1gpu_8epochs_lr2e4_bs8_fp16_linear.log 2>&1
78
  ```
79
 
 
 
 
 
 
 
 
 
 
 
 
80
  ## Training Data:
81
 
82
  * https://huggingface.co/datasets/FinGPT/fingpt-sentiment-train
 
77
  > train_internlm-20b_1gpu_8epochs_lr2e4_bs8_fp16_linear.log 2>&1
78
  ```
79
 
80
+ ## inference script
81
+ ```
82
+ CUDA_VISIBLE_DEVICES=1 python benchmarks.py \
83
+ --dataset fpb,fiqa,tfns,nwgi \
84
+ --base_model internlm-20b \
85
+ --peft_model FinGPT/fingpt-sentiment_internlm-20b_lora \
86
+ --batch_size 1 \
87
+ --max_length 512 \
88
+ --from_remote True
89
+ ```
90
+
91
  ## Training Data:
92
 
93
  * https://huggingface.co/datasets/FinGPT/fingpt-sentiment-train