最近学习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 年前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐