Commit dd3a55ad authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Output predictions and metrics in train mode for multihead. 

This is to be consistent with other heads who output the predictions and metric tensors in train mode. Outputting the metric tensors allow us for example to plot the metrics on the training set (and compare them to the metircs on the eval set).

PiperOrigin-RevId: 168564791
parent 92362d0f
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment