ModelArts Autosearch 训练任务用例:强化学习rllib trainer

举报
7corner7 发表于 2020/09/11 18:14:24 2020/09/11
【摘要】 本案例提供利用 ModelArts 上的 Autosearch 训练作业服务实现对强化学习算法库 rllib trainer 的超参搜索的一些配置方案。超参搜索配置(autosearch_config)以 bayes 超参搜索为例(注意 bayes要求连续性搜索空间)简易强化学习训练脚本(bootstrap_file)示例注意配置 ray 的训练环境:autosearch 本身镜像里的 Te...

本案例提供利用 ModelArts 上的 Autosearch 训练作业服务实现对强化学习算法库 rllib trainer 的超参搜索的一些配置方案。


  1. 超参搜索配置(autosearch_config)

    以 bayes 超参搜索为例(注意 bayes要求连续性搜索空间)

    image.png

  2. 简易强化学习训练脚本(bootstrap_file)示例

    image.png

  3. 注意配置 ray 的训练环境:autosearch 本身镜像里的 TensorFlow 版本过低,安装环境时除了安装 gym, ray 或 rllib 外,还需要重新安装 TensorFlow(如本例重安装 tf==1.13匹配 ray==0.8.0)

    image.png

    若是云道环境,直接重新安装可能会引发 protobuf 版本错误,建议用以下代码安装

    image.png

  4. 通过以上配置,可以调用 autosearch 服务进行超参搜索,各 trials 结果存储在 {train_url}/log/trials.csv 中。




【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。