Plotting multiple metrics in a single graph

[abhinavg97]

🚀 Feature

Can we have multiple metrics plotted on the same graph in Tensorboard logging done by lightning?
That is plotting the dictionary values returned in log_dict in the same graph.

Motivation

Pitch

Alternatives

Additional context

[github-actions]

Hi! thanks for your contribution!, great first issue!

[SkafteNicki]

You can already do this in lightning, by calling the tensorboard logger yourself, using the add_scalars method.
If you insert this into your code (artificial example)

        self.logger.experiment.add_scalars('main_tag', 
                                           {'metric1': loss, 
                                            'metric2': loss+1}, 
                                           global_step=self.global_step)

it will produce this in tensorboard

where the metrics are shown in the same plot.

[abhinavg97]

Thank you so much!