deepke使用过程问题汇总

deepke是一个知识图谱处理框架,可以方便的进行实体抽取,关系提取,属性抽取,里面有各种提取模块,包括bert,lstmcrf等,可以方便的进行各种任务处理。

但是在使用的过程中,也有很多问题需要注意。我在处理红楼梦实体抽取的时候,也是遇到各种问题,在此记录一下。

1,没有GPU时在进行ner训练时报错

if USE_MULTI_GPU and torch.cuda.device_count() > 1:
    MULTI_GPU = True
    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
    n_gpu = torch.cuda.device_count()

此处如果没走到,在后面使用n_gpu时会调用n_gpu未定义,需要else增加默认值

2,wandb: WARNING W&B installed but not logged in. Run wandb login or set the WANDB_API_KEY env variable

此处可以选择3,不用日志,或者直接将wandb代码删除

3,ImportError: No module named 'ipdb'

使用conda 创建python环境

你可能感兴趣的:(人工智能,深度学习)