grid search python_Ray tune:一个可扩展的超参数优化Python库

Ray tune:一个可扩展的超参数优化Python库

Tune是一个超参数优化库,可以用于PyTorch、TensorFlow, MXnet,keras等深度学习框架


超参数优化

在深度学习中, 除了可以学习参数外,还存在很多超参数,这些超参数对于网络的性能影响也十分巨大,不同的机器学习任务往往需要不 同的超参数,常见的超参数有: - 网络结构,包括神经元的连接关系,层数,每层的神经元的数量、激活函数的类型 - 优化参数,包括优化方法,学习率,小批量的样本数量等 - 正则化系数

对于超参数的搜索,一般使用的方法:人工搜索网格搜索随机搜索贝叶斯优化

动态资源分配

由于目前深度学习的优化方法一般都采取随机梯度下降,因此我们可以通 过一组超参数的学习曲线来预估这组超参数配置是否有希望得到比较好的结 果。如果一组超参数配置的学习曲线不收敛或者收敛比较差,我们可以应用早 期停止(early-stopping)策略来中止当前的训练。

Tune的核心特征

  1. 多计算节点的分布式超参数的查找
  2. 支持多种深度学习框架,例如:pytorch,TensorFlow
  3. 结果直接可以用tensorboard可视化
  4. 可拓展的SOTA算法,例如:PBT,HyperBand/ASHA
  5. 整合了很多超参数优化库, 例如:Ax, HyperOpt,Bayesian Optimization

快速开始

安装下列工具:

pip install ray torch torchvison filelock

实例是使用网格搜索来训练一个CNN模型

import 

使用Tensorboard可视化实验结果:

tensorboard --logdir ray_results

v2-c2513a1fa2ffc60259ed547533715a78_b.jpg
参考:
Tune: A Scalable Hyperparameter Tuning Library​ray.readthedocs.io
v2-05d9ddf18e9fbfcc5e16a36e5ef49e34_180x120.jpg
Ray tune:一个可扩展的超参数优化Python库 - Neolei | Lei's Blog​neolei.github.io
v2-a9add337ee39d79878975f2c2d0a1c43_ipico.jpg
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值