Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
XuanYuan_pytorch
Commits
ba499475
Commit
ba499475
authored
Jul 03, 2024
by
wanglch
Browse files
Update README.md
parent
b77a7567
Pipeline
#1306
canceled with stages
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
6 deletions
+4
-6
README.md
README.md
+4
-6
No files found.
README.md
View file @
ba499475
...
@@ -88,7 +88,7 @@ pip install -e ".[torch,metrics]"
...
@@ -88,7 +88,7 @@ pip install -e ".[torch,metrics]"
## 数据集
## 数据集
迷你数据集
[
fingpt_sentiment
](
./
LLaMA-Factory/
data/fingpt_sentiment.json
)
迷你数据集
[
fingpt_sentiment
](
./data/fingpt_sentiment.json
)
预训练需要准备你的训练数据,需要将所有样本放到一个列表中并存入json文件中。每个样本对应一个字典,包含以下信息,示例如下所示:用于正常训练的完整数据集请按此目录结构进行制备:
预训练需要准备你的训练数据,需要将所有样本放到一个列表中并存入json文件中。每个样本对应一个字典,包含以下信息,示例如下所示:用于正常训练的完整数据集请按此目录结构进行制备:
...
@@ -103,8 +103,6 @@ pip install -e ".[torch,metrics]"
...
@@ -103,8 +103,6 @@ pip install -e ".[torch,metrics]"
## 训练
## 训练
训练需要使用Llama-Factory包,需要将本仓库中Llama-Factory的data数据替换git clone的Llama-Factory的data文件,替换后只保留git clone的Llama-Factory文件夹。
根据实际情况在脚本中修改权重相关路径
根据实际情况在脚本中修改权重相关路径
### 单机多卡
### 单机多卡
...
@@ -130,7 +128,7 @@ sh Xuanyuan_inference.sh
...
@@ -130,7 +128,7 @@ sh Xuanyuan_inference.sh
### 精度
### 精度
测试数据:
[
fingpt_sentiment
](
./
LLaMA-Factory/
data/fingpt_sentiment.json
)
,使用的加速卡:K100。
测试数据:
[
fingpt_sentiment
](
./data/fingpt_sentiment.json
)
,使用的加速卡:K100。
| device | train_loss | eval_loss
| device | train_loss | eval_loss
| :------: | :------: | :------: |
| :------: | :------: | :------: |
...
@@ -139,10 +137,10 @@ sh Xuanyuan_inference.sh
...
@@ -139,10 +137,10 @@ sh Xuanyuan_inference.sh
## 应用场景
## 应用场景
### 算法类别
### 算法类别
`问答`
`
对话
问答`
### 热点应用行业
### 热点应用行业
`金融,教育`
`金融,教育
,银行
`
## 预训练权重
## 预训练权重
-
[
Duxiaoman-DI/XuanYuan-13B-Chat
](
https://modelscope.cn/models/Duxiaoman-DI/XuanYuan-13B-Chat/files
)
-
[
Duxiaoman-DI/XuanYuan-13B-Chat
](
https://modelscope.cn/models/Duxiaoman-DI/XuanYuan-13B-Chat/files
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment