文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

极简可视化工具Aim发布,跑得比TensorBoard快多了

2024-12-03 17:20

关注

来自加州伯克利的团队开源了Aim,一个号称搜索速度比TensorBoard快好几倍的机器学习可视化工具包,在Reddit上成为高热话题。

Aim可以在几分钟内记录、搜索和比较100项实验,而在TensorBoard或MLFlow上进行大量实验比较可能需要花费数小时。这对于实验管理非常有用,而且Aim超级容易上手。

和TensorBoard/MLFlow相比,Aim的优点主要是支持:

比如,我们只想看训练集上的试验结果,将context.subset设置为train:

如果还想在搜索结果的基础上做进一步的筛选,去掉学习率0.00001的部分,只需再加入一个and条件:

如果对图中某个数据点感兴趣,点击它即可快速定位到相应表格位置。

是不是很方便?虽然界面可能简陋了点,但简洁易用就是Aim的主要特色。有网友表示,Aim看起来非常整洁,搜索语言看起来也很易用。

近年来,AI实验方面诞生了像trains和wandb这类第三方可视化工具。与这类工具比较,Aim在速度和数据隐私方面有很大的优势。

运行Aim需要安装Docker,Aim本身通过pip方式安装。

pip3 install aim-cli

输入以下命令即可运行Aim的UI:

aim up

前提是你要在自己的AI模型里导入Aim

import aim…aim.set_params(hyperparam_dict, name=’params_name’)aim.track(metric_value, name=’metric_name’, epoch=the_epoch_value)…

Aim提供的方法有跟踪数据(track)、设置超参数(set_params)、指定自定义目录(session)。

最近,作者又给Aim加入了一项新功能:支持使用Tensorboard日志。方法如下:

aim up --tf_logs path/to/logs

此命令将在TensorFlow摘要日志上启动Aim,并从给定路径递归加载日志。

Aim的开发者承诺未来将提供Pytorch Lightning和Keras集成。

GitHub地址:
https://github.com/Aimhubio/Aim

 

 

来源:量子位内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯