BERT模型训练GPU,预测走cpu
bert
TensorFlow code and pre-trained models for BERT
项目地址:https://gitcode.com/gh_mirrors/be/bert
免费下载资源
·
最近学习BERT,训练用GPU,预测默认也走GPU,比较耗费资源,启用GPU加载也会有很多信息,想用CPU,设定如下就好:
预测代码之前加入os.environ["CUDA_VISIBLE_DEVICES"]="-1"
get了,game over。
GitHub 加速计划 / be / bert
8
2
下载
TensorFlow code and pre-trained models for BERT
最近提交(Master分支:4 个月前 )
eedf5716
Add links to 24 smaller BERT models. 4 年前
8028c045 - 4 年前
更多推荐
已为社区贡献8条内容
所有评论(0)