【文件属性】:
文件名称:neurst:神经端到端语音翻译工具包
文件大小:1022KB
文件格式:ZIP
更新时间:2021-03-20 09:42:44
Python
NeurST:神经语音翻译工具包
NeurST旨在轻松构建和培训端到端语音翻译,该语言具有精心设计的可扩展性和可伸缩性。我们相信,这种设计可以使NLP研究人员更容易入门。此外,NeurST允许研究人员训练定制模型以进行翻译,摘要等。
NeurST基于TensorFlow2,我们正在开发pytorch版本。
特征
楷模
NeurST提供各种模型的参考实现,包括:
变压器(自我关注)网络
即将推出...
食谱和基准
NeurST为各种任务提供了几个强大且可重复的基准测试:
翻译
语音转文字
此外
一台机器或多台机器上的多GPU(分布式)培训
MirroredStrategy / MultiWorkerMirroredStrategy
/
混合精度训练(训练速度更快,GPU内存更少)
实现了多种搜索算法:
光束搜索
采样(无约束,top-k和top-p)
即使在单个GPU上也可以通过延迟更