新智元报道
来源:reddit等
编辑:科雨、白峰
【新智元导读】还在抱怨模型训练过于耗时?还在手动苦苦调整超参?现在,这款神器来帮你!24倍分布式训练加速,100倍智能超参优化,Determind AI宣布开源,你的模型有救了!
作为一名深度学习科研人员,小编我可是深受模型训练的困扰:一个模型要跑几天几夜,结果最后发现有一步错了 ...从头再来
超参数怎么调精度都上不去,心真的好累...
手动调整系统环境,只要错一个,就跑不起来了嗷...
常常只是为了训练一个模型,就要耗费巨大的时间,改来改去,还有经过漫长的等待,内心真的在默默哭泣有木有!
然后,我发现我竟然还能再拯救一下——Determined AI驾着七彩祥云来救我啦!
同样受到这种苦恼,深有感触的众多深度学习研究人员,花费了数年心血,终于开发出了一个便利广大深度学习从业人员的训练神器。
这款深度学习训练平台,是由多年从事实践领域的专家,耗费三年时间构建完成的,他们的目标是帮助深度学习团队更快地训练模型,轻松共享GPU资源并有效协作。
Determined使深度学习工程师可以集中精力大规模构建和训练模型,而无需担心DevOps,或者为常见任务(如容错或实验跟踪)编写代码。
官网链接:
https://determined.ai/developers/
“等一下,这里面到底有啥,能不能详细说明一下!”
好,那我就好好扒一扒,给大家看看。
专注模型训练,更快更准!
众所周知,一个完整的深度学习项目,包括了数据准备,模型训练以及模型部署等环节:
而Determined,则可以帮助我们更好的专注于模型本身的训练,而不是把时间和精力大量花在样例代码和DevOps上。
模型训练部分,Determined可以帮助我们完成:
- 更快的分布式训练
- 智能的超参优化
- 实验跟踪和可视化
这样,它就可以使我们可以专注于手头的任务——即训练模型上。
训练人员,可以立即进入为深度学习工作而创建的专用环境,然后将时间花费在来设置模型上,而完全不必担心安装,拆卸和其他样例代码的问题。
看到这里我想说,开发人员真的是太贴心了,他们很清楚我们不想花时间做什么,我真的感激*100!
那这个神器可以帮我们处理什么呢?
- 内置的训练循环抽象,可支持实验跟踪,有效的数据加载,容错,并可以灵活地进行自定义。
- 高性能的分布式培训,无需更改任何代码。
- 基于前沿研究的自动超参数优化。
有了这些功能,模型训练so easy!老板再也不用担心我的模型!
直接点击官网链接,就可以体验这些功能:
接下来,我们可以看看这两个和模型训练密切相关的部分——分布式训练速度和智能超参优化具体是怎么一回事:
分布式训练
Determined主要运用了Horovod,以Horovod为起点,研究人员运用了多年的专业知识和经验,使得整个训练过程比库存配置要快得多。
这样,我们就可以更减少浪费的时间,更好的利用计算机的硬件设备,来达到高效的训练速度。
在这里科普一下Horovod:
Horovod 是一套面向TensorFlow 的分布式训练框架,由Uber 构建并开源,目前已经运行于Uber 的Michelangelo 机器学习即服务平台上。Horovod 能够简化并加速分布式深度学习项目的启动与运行。当数据较多或者模型较大时,为提高机器学习模型训练效率,一般采用多 GPU 的分布式训练。TensorFlow 集群存在诸多缺点,如概念太多、学习曲线陡峭、修改的代码量大、性能损失较大等,而 Horovod 则让深度学习变得更加美好,随着规模增大,Horovod 性能基本是线性增加的,损失远小于 TensorFlow。
有关Horovod的详细信息,大家可以点击下方链接进行学习和了解~
"是时候放弃 TensorFlow 集群,拥抱 Horovod 了"
https://www.infoq.cn/article/J4ry_9bsfbcNkv6dfuqC
使用Determined,我们的分布式模型训练速度能有多快呢?下图告诉我们,可以达到24倍!!
PS:真的有快很多啊噜,感慨...
超参数优化
通过提供和任务密集集成的前沿智能搜索功能,以及默认并行的设置,开发者帮我们免除了传统超参搜索调试的麻烦。
这样,我们就可以用更快的速度,来获得更准确的模型。
那使用Determined,超参数优化的速度又可以提高多少倍呢?
100倍!看到这里,我真的很真实的在感动了... 多出的时间是不是可以用来玩动森? 反正也有正当理由:为ACAL2020做准备哈哈~
GPU调度更加灵活,支持各种流行框架
Determined还可以更加灵活的进行GPU调度,包括:
1)动态调整训练任务的大小
2)自动在AWS和GCP上管理云资源
同时,它还可以支持TensorBoard和基于GPU的Jupyter Notebook。(我最亲爱的小伙伴也可以在这个平台上使用了诶!!)
“Tensorflow和Pytorch都能支持吗?”——吃瓜的路人甲忍不住了
哦对了,还有一个关键点,那就是使用这个平台的同时,我们可以继续使用Tensorflow和Pytorch,只要改一下模型的代码,来实现Determined的API就可以了。
是不是很方便!
平台透明开放,用户服务完备
深度学习团队协作工具
这个平台透明,开放,并且方便团队协作,很适合深度学习团队使用:
Determined通过实验跟踪,日志管理,指标可视化,可重复性和依赖性管理等,帮助任何人在实验管理中脱颖而出。对任何规模的团队来说,这款平台都是完美的。
从一个团队开始,使用者可以在准备就绪后,轻松地共享和扩展工作。
开源,适用于各种云供应商
开发者建议:深度学习从业人员可以选择跳出专有解决方案:
“Determined可以支持云平台,或者本地基础结构,甚至两者都可以。该平台可以在您选择的深度学习框架中工作,并随时导出到流行的服务框架中。”
使用这个平台,使用非常简单的步骤,就可以开源项目,从而造福其他小伙伴~
详细的用户文档
Determined现在已经有正式的用户手册,可以供使用者随时查阅和参考:
https://docs.determined.ai/latest/
还有简单容易上手的tutorial:
最后,开发者还特别贴心的附上了Slack链接,任何人都可以和Determined的开发团队交流,并进行反馈和沟通:
这里放上项目开源地址,有兴趣的小伙伴可以马上去GitHub体验下:
https://github.com/determined-ai/determined
参考资料:
https://www.reddit.com/r/MachineLearning/comments/gaecbq/n_determined_deep_learning_training_platform/
https://determined.ai/developers/