Update README.md
Browse files
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
|