How to log training and validation on the same plot (upgrading from 1.7.7 to 2.2.0+post0)

I want to log my training and validation metrics on the same plot as two lines of different colors. I used to do that using torch lightning 1.7.7 with the following code:

self.log_dict({'output_1 Loss': {'VALIDATION': 0}, 'output_2 Loss': {'VALIDATION': 0}})

After updating to 2.2.0+post0, I get the error:

ValueError: `self.log(output_1 Loss, {'VALIDATION': 0})` was called, but `dict` values cannot be logged

How should I edit my code?

Hi @mfoglio, I am also facing the same issue have you found a solution?