Skip to content

Enable Rich Progress Bar/Model Summary if Rich is available #9580

Open
@SeanNaren

Description

@SeanNaren

🚀 Feature

Currently we have the Rich Progress Bar/Summary in Lightning. However, we only enable it if the user passes the Rich Progress Bar Callback to the Trainer manually.

I see a few options, based on what we'd like to go for in the future:

  1. Enable the Rich Progress Bar/Model Summary if Rich is available (@ananthsub mentioned that we might need to check for Rich 10+ Add RichModelSummary callback #9546 (comment))
  2. Put the Rich Progress Bar/Summary in Bolts. This means we can only enable the Rich Progress Bar if Bolts is installed (and a version that contains the progress bar)
  3. Keep as is

Personally I'm a big fan of 1 as it gives users a nice new progress bar/model summary to look at, which would be nice for a new release! All the user needs to do is pip install rich -U.

If you have other alternatives please put below! Also please let me know which options you'd like

cc @Borda @justusschock @kaushikb11 @rohitgr7

Metadata

Metadata

Assignees

No one assigned

    Labels

    featureIs an improvement or enhancementgood first issueGood for newcomersplGeneric label for PyTorch Lightning packageprogress bar: rich

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions