PyTorch Lightning Introduces Improved Lightning Logger Support for Weight & Biases and

The Lightning Team recently announced that it has collaborated closely with and Weight & Biases Teams to provide the best experience in terms of Logging and Experiment Tracking.

The Lightning Team recently announced that it has collaborated closely with and Weight & Biases Teams to provide the best experience in terms of Logging and Experiment Tracking. The launch of PyTorch Lightning v1.5 marked a major leap in reliability to support the increasingly complex demands of the leading AI organizations and prestigious research labs that rely on Lightning to develop and deploy AI at scale. To better support the fast-growing community, PyTorch Lightning aims at becoming the simplest, most flexible framework for expediting any kind of deep learning research to production.

Lightning 1.5 adds new methods to WandbLogger that help you elevate your logging experience inside PL by giving you the ability to monitor your model weights and give you the functionality to log other artifacts such as text, tables, images, and model checkpoints. By using watch and unwatch methods, you can track gradients, parameter histogram, model topology and it can be used directly within the LightningModule. 

Image Source: PyTorch Lightning

But the most exciting feature is the support of log_table directly though the WandbLogger.

Subscribe to our Newsletter

Join our editors every weekday evening as they steer you through the most significant news of the day, introduce you to fresh perspectives, and provide unexpected moments of joy
Your newsletter subscriptions are subject to AIM Privacy Policy and Terms and Conditions.
Image Source: PyTorch Lightning

Below is an example of how W&B tables can be used to provide insightful reports.

Image Source: PyTorch Lightning

With the v1.5 release, NeptuneLogger is revamped and relies on the new improved neptune-client.

To utilize the NeptuneLogger with Lightning, simply do the following:

Image Source: PyTorch Lightning

Using the NeptuneLogger, you can log images, texts, and artifacts.

Image Source: PyTorch Lightning

And using the log_model_summary method, you can track the summary of the parameters of your models.

Image Source: PyTorch Lightning

The Lightning Team seems more than ever committed to provide users with the best experience possible to anyone doing optimization with PyTorch and the PyTorch Lightning API being already stable, breaking changes will be minimal. You can learn more though the WandbLogger documentation and the NeptuneLogger documentation.

Victor Dey
Victor is an aspiring Data Scientist & is a Master of Science in Data Science & Big Data Analytics. He is a Researcher, a Data Science Influencer and also an Ex-University Football Player. A keen learner of new developments in Data Science and Artificial Intelligence, he is committed to growing the Data Science community.

Download our Mobile App


AI Hackathons, Coding & Learning

Host Hackathons & Recruit Great Data Talent!

AIM Research

Pioneering advanced AI market research

Request Customised Insights & Surveys for the AI Industry

The Gold Standard for Recognizing Excellence in Data Science and Tech Workplaces

With Best Firm Certification, you can effortlessly delve into the minds of your employees, unveil invaluable perspectives, and gain distinguished acclaim for fostering an exceptional company culture.

AIM Leaders Council

World’s Biggest Community Exclusively For Senior Executives In Data Science And Analytics.

3 Ways to Join our Community

Telegram group

Discover special offers, top stories, upcoming events, and more.

Discord Server

Stay Connected with a larger ecosystem of data science and ML Professionals

Subscribe to our Daily newsletter

Get our daily awesome stories & videos in your inbox

Subscribe to Our Newsletter

The Belamy, our weekly Newsletter is a rage. Just enter your email below.