Aim v2.2.0 — Hugging Face integration

Aim 2.2.0 is out! Thanks to the incredible Aim community for the feedback and support on building democratized open-source AI dev tools.

Thanks to siddk, TommasoBendinelli and Khazhak for their contribution to this release.

Note on the Aim versioning: The previous two release posts: Aim 1.3.5 and Aim 1.3.8 had used the version number of AimUI as those contained only UI changes. From now on we are going to stick to the Aim versions only regardless of the type of changes to avoid any confusion. Check out the Aim CHANGELOG.

We have also added milestones for each version. As well as the Roadmap.

Check out the new features at play.aimstack.io.

Hugging Face integration

Hugging Face integration has been one of the most requested features so far and we are excited to finally ship it.

Hugging Face and Aim

Here is a code snippet on how easy it is to integrate Aim and Hugging Face.

Here is how it works: aim_callback will automatically open an Aim Session and close it when the trainer is done. When trainer.train(), trainer.evaluate() or trainer.predict() is called, aim_callback will automatically log the hyper-params and respective metrics.

Find a full example here.

Metric Visibility Control

When dealing with lots of experiments some metrics may need to be hidden (while still being in the Search) to allow better visibility of the overall picture.

A toggle is now available both for each metric/row as well as for global on/off. Check out the demo below:

Hide individual metrics as well as collectively

Column resize

With lots of params tracked, some of them are too wide for table and take over the whole screen real estate.

Column resize will allow to fully control data width on the table. Resize is available both on Explore and Dashboard tables. Here is a quick demo:

Drag column edges back-and-forth to resize

Hide columns with similar values

More often than not params have lots of repetition. Especially when 100s of them are tracked. In that case the explore table becomes super-noisy.

This feature will allow only the relevant info to be shown. This has been the missing piece of the column management that many had requested. Here is a quick demo:

Leave only different columns with a button click, if needed customize afterwards

Logscale

One of the must-have features that we hadn’t had a chance to work on. Finally it’s available!!

log-scale on Aim

New methods for aggregation

Now you can select different types of aggregations as well as whether to see the whole area or just the aggregated lines.

Learn More

Aim is on the mission to democratize AI dev tools.

We have been incredibly lucky to get help and contributions from the amazing Aim community. It’s humbling and inspiring.

Try out Aim, join the Aim community, share your feedback.

And don’t forget to leave Aim a star on GitHub for support 🙌.

This article was originally published on Aim Blog. Find more in depth guides and details of the newest releases there.