Training UI - Multiple output layers

DL4J
1

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:

Unbenannt
Unbenannt450×849 32.4 KB

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.

2

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.