首页|期刊导航|北京大学学报(自然科学版)|N3LDG:一种轻量级自然语言处理深度学习库

N3LDG:一种轻量级自然语言处理深度学习库OA北大核心CSCDCSTPCD

N3LDG: A Lightweight Neural Network Library for Natural Language Processing

中文摘要

提出一种用于自然语言处理的轻量级深度学习库N3LDG,可以支持动态地构建计算图,并能自动地批量化执行计算图.实验显示,当训练卷积神经网络、双向LSTM和树结构LSTM时,N3LDG都能高效地构建与执行计算图;当使用CPU训练上述模型时,N3LDG的训练速度优于PyTorch;当使用GPU训练卷积神经网络和树结构LSTM模型时,N3LDG的训练速度优于PyTorch.

王潜升;余南;张梅山;韩子嘉;付国宏

黑龙江大学计算机科学技术学院,哈尔滨 150080黑龙江大学计算机科学技术学院,哈尔滨 150080黑龙江大学计算机科学技术学院,哈尔滨 150080黑龙江大学计算机科学技术学院,哈尔滨 150080黑龙江大学计算机科学技术学院,哈尔滨 150080

深度学习库自然语言处理轻量级CUDA

《北京大学学报(自然科学版)》 2019 (1)

113-119,7

国家自然科学基金(61672211, 61602160)和黑龙江省自然科学基金(F2016036)资助

10.13209/j.0479-8023.2018.065

评论

您当前未登录!去登录点击加载更多...