Training UI - Multiple output layers

I have a computation graph that has two ouput layers. I am using the sample code from

Here is my model architecture:

And I want to see the Model Score vs Iteration graph of policy and value head as seperate charts because they use different losses. But at the moment it only shows the policy loss as score.

The loss it shows is the overall network loss, which is an average of the two losses.

Unfortunately, at the moment the UI does not support displaying multiple separate losses.