博客
关于我
跑得比TensorBoard快多了,极简可视化工具Aim发布 | Reddit高热
阅读量:241 次
发布时间:2019-03-01

本文共 1269 字,大约阅读时间需要 4 分钟。

晓查 发自 凹非寺 
量子位 报道 | 公众号 QbitAI

来自加州伯克利的团队开源了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

本文系网易新闻•网易号特色内容激励计划签约账号【量子位】原创内容,未经账号授权,禁止随意转载。

榜单征集!7大奖项锁定AI TOP企业

「2020中国人工智能年度评选」报名中!评选将从公司、人物、产品、社区四大维度共7个奖项寻找优秀的AI企业,欢迎大家扫码报名参与。 

评选报名截止于11.20,并将于12月16日揭晓,也期待与百万从业者们,共同见证这些优秀企业的荣誉!

量子位 QbitAI · 头条号签约作者

վ'ᴗ' ի 追踪AI技术和产品新动态

一键三连「分享」、「点赞」和「在看」

科技前沿进展日日相见~

你可能感兴趣的文章
MySql中mvcc学习记录
查看>>
mysql中null和空字符串的区别与问题!
查看>>
MySQL中ON DUPLICATE KEY UPDATE的介绍与使用、批量更新、存在即更新不存在则插入
查看>>
MYSQL中TINYINT的取值范围
查看>>
MySQL中UPDATE语句的神奇技巧,让你操作数据库如虎添翼!
查看>>
Mysql中varchar类型数字排序不对踩坑记录
查看>>
MySQL中一条SQL语句到底是如何执行的呢?
查看>>
MySQL中你必须知道的10件事,1.5万字!
查看>>
MySQL中使用IN()查询到底走不走索引?
查看>>
Mysql中使用存储过程插入decimal和时间数据递增的模拟数据
查看>>
MySql中关于geometry类型的数据_空的时候如何插入处理_需用null_空字符串插入会报错_Cannot get geometry object from dat---MySql工作笔记003
查看>>
mysql中出现Incorrect DECIMAL value: '0' for column '' at row -1错误解决方案
查看>>
mysql中出现Unit mysql.service could not be found 的解决方法
查看>>
mysql中出现update-alternatives: 错误: 候选项路径 /etc/mysql/mysql.cnf 不存在 dpkg: 处理软件包 mysql-server-8.0的解决方法(全)
查看>>
Mysql中各类锁的机制图文详细解析(全)
查看>>