用上 Pytorch Lightning 这六招,深度学习 pipeline 可提速 10 倍

用上 Pytorch Lightning 这六招,深度学习 pipeline 可提速 10 倍

面对数以亿计的图片数据,到底该用什么样的方法才能快速搞实验?这样的问题,或许在做机器学习研究的你,也会经常遇到。而就在最近,一个国外小哥就提出了一种建议:在 Pytorch lightning 基础上,让深度学习 pipeline 速度提升 10 倍!用他自己的话来说就是 ——“爬楼时像给了你一个电梯”。这般“酸爽”,到底是如何做到的呢?优化机器学习 pipeline,很重要无论你是身处学术界还是工业界,时间和资源等各种因素,往往会成为你在搞实验的枷锁。尤其是随着数据集规模和机器学习模型,变得越...