SkyID
/
deepke
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug

This commit is contained in:
tlk-dsg 2021-12-11 17:49:29 +08:00
parent 548ca915b6
commit 1295c19b33
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example/ner/few-shot/conf/train/few_shot.yaml
View File

@ -1,6 +1,6 @@
seed: 1 seed: 1
bart_name: "/data/lilei/project/BARTNER-AMAX/facebook/bart-large" bart_name: "facebook/bart-large"
dataset_name: mit-movie dataset_name: mit-movie
device: cuda device: cuda

4
setup.py
View File

@ -1,7 +1,7 @@
from setuptools import setup, find_packages from setuptools import setup, find_packages
setup( setup(
name='deepke', # 打包后的包文件名 name='deepke', # 打包后的包文件名
version='0.2.92', #版本号 version='0.2.93', #版本号
keywords=["pip", "RE","NER","AE"], # 关键字 keywords=["pip", "RE","NER","AE"], # 关键字
description='DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。', # 说明 description='DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。', # 说明
long_description="client", #详细说明 long_description="client", #详细说明
@ -25,7 +25,7 @@ setup(
'seqeval==1.2.2', 'seqeval==1.2.2',
'tqdm==4.60.0', 'tqdm==4.60.0',
'opt-einsum==3.3.0', 'opt-einsum==3.3.0',
'wandb==0.12.7' 'wandb==0.12.7',
"ujson" "ujson"
], ],
classifiers=[ classifiers=[