I have a computation graph that has two ouput layers. I am using the sample code from https://deeplearning4j.org/docs/latest/deeplearning4j-nn-visualization.
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.