当前栏目
极简可视化工具Aim发布,跑得比TensorBoard快多了
本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。
来自加州伯克利的团队开源了Aim,一个号称搜索速度比TensorBoard快好几倍的机器学习可视化工具包,在Reddit上成为高热话题。
![极简可视化工具Aim发布,跑得比TensorBoard快多了](https://s6.51cto.com/oss/202011/09/89f492b9516eb340ef945b1b3868ccbd.png)
Aim可以在几分钟内记录、搜索和比较100项实验,而在TensorBoard或MLFlow上进行大量实验比较可能需要花费数小时。这对于实验管理非常有用,而且Aim超级容易上手。
![极简可视化工具Aim发布,跑得比TensorBoard快多了](https://s6.51cto.com/oss/202011/09/b03e78368239e133909a4374b4ce872d.png)
和TensorBoard/MLFlow相比,Aim的优点主要是支持:
- 按参数进行搜索、分组
- 分列图表
- 汇总大量实验查看趋势
- 其他较小的实验指标和参数操作
比如,我们只想看训练集上的试验结果,将context.subset设置为train:
![极简可视化工具Aim发布,跑得比TensorBoard快多了](https://s6.51cto.com/oss/202011/09/657bc866d9699dc99c5f960f917e8b61.gif)
如果还想在搜索结果的基础上做进一步的筛选,去掉学习率0.00001的部分,只需再加入一个and条件:
![极简可视化工具Aim发布,跑得比TensorBoard快多了](https://s6.51cto.com/oss/202011/09/69e0841bd805a0009c4024af3f2c5f96.gif)
如果对图中某个数据点感兴趣,点击它即可快速定位到相应表格位置。
![极简可视化工具Aim发布,跑得比TensorBoard快多了](https://s3.51cto.com/oss/202011/09/270613ff4e8bb931d2dff594dc2e0e4b.gif)
是不是很方便?虽然界面可能简陋了点,但简洁易用就是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
相关文章
- 鲜为人知但很有用的 HTML 属性
- 翻转再翻转!有意思的水平横向溢出滚动
- 自定义计数器小技巧!CSS 实现长按点赞累加动画
- 过五关!React高频面试题指南
- 软件开发中的十个认知偏差
- 不需要 JS!仅用 CSS 也能达到监听页面滚动的效果!
- 一文读懂TypeScript类型兼容性
- Vue 的响应式原则与双向数据绑定
- 快速掌握 TypeScript 新语法:Infer Extends
- JWT教你如何证明你是我的人!
- 一篇带给你 V8 GC 的实现
- 面试官:请使用JS完成一个LRU缓存?
- 通过可视化来学习JavaScript事件循环
- 新的跨域策略:使用 COOP、COEP 为浏览器创建更安全的环境
- 为什么有人说 vite 快,有人却说 vite 慢?
- 种草 Vue3 中几个好玩的插件和配置
- 超全面的前端工程化配置指南
- Vue 状态管理未来样子
- Volatile关键字能保证原子性么?
- 面试突击:SpringBoot 有几种读取配置文件的方法?