语言模型代码中强制获取GUP
Created by: xiangyubo
为使您的问题得到快速解决,在建立Issue前,请您先通过如下方式搜索是否有相似问题:【搜索issue关键字】【使用labels筛选】【官方文档】
建立issue时,为快速解决问题,请您根据使用情况给出如下信息:
- 标题:简洁、精准描述您的问题,例如“ssd 模型前置lstm报错 ”
- 版本、环境信息: 1)PaddlePaddle版本:1.5
- 模型信息 1)模型名称:语言模型
在外层 train.py 中有判断当前环境是否为GPU环境,但是在模型定义中,强制使用了 fluid.core.get_cuda_device_count() 函数,导致 CPU 环境运行报错。
下列代码的 257行 https://github.com/PaddlePaddle/models/blob/develop/PaddleNLP/models/language_model/lm_model.py