Force enable tqdm in Google Colab

🚀 Feature

Recently, Google Colab deactivated tqdm progress bars in Jupyter notebooks. We currently get this message from Google:

Motivation

Users don’t see a progress bar despite our docs saying that it is enabled by default.

Pitch

Force enable the tqdm widgets when in Colab. We can detect whether the code is running in Google Colab.

from google.colab import output
output.enable_custom_widget_manager()

Alternatives

x

Additional context

Users reported this on Slack for the first time.

---

If you enjoy Lightning, check out our other projects! ⚡

• Metrics: Machine learning metrics for distributed, scalable PyTorch applications.

• Lite: enables pure PyTorch users to scale their existing code on any kind of device while retaining full control over their own loops and optimization logic.

• Flash: The fastest way to get a Lightning baseline! A collection of tasks for fast prototyping, baselining, fine-tuning, and solving problems with deep learning.

• Bolts: Pretrained SOTA Deep Learning models, callbacks, and more for research and production with PyTorch Lightning and PyTorch.

• Lightning Transformers: Flexible interface for high-performance research using SOTA Transformers leveraging PyTorch Lightning, Transformers, and Hydra.

cc @tchaton @rohitgr7 @awaelchli

[kaushikb11]

@awaelchli TQDMProgressBar seems to be working now on Colab.

Could you confirm? Notebook

[awaelchli]

@kaushikb11 You are right. It looks like the problem has solved itself over night. I opened a fresh Colab notebook and the trainer produced the expected progress bars.