GradientFlow:90秒训练ImageNet/AlexNet的分布式通讯库

来自商汤和南洋理工的研究人员,实现了仅用1.5分钟训练ImageNet/AlexNet。尽管多种物体识别方法在ImageNet上已经达到了很高的准确率,但模型训练过程仍然非常耗时,缩短模型训练时间是AI研究人员必须面临的工程挑战。为减少训练时间,作者开发了用于分布式数据通讯后端GradientFlow,辅以一系列的网络通讯优化技巧实现惊人的效率提升。 论文地址: https://arxiv.org
相关文章
相关标签/搜索