Commit be9e0981 authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

TFTS: Don't return model state from OneShotPredictionHead evaluate() 

It doesn't need to be fed (OneShot), and returning it as a tuple causes some issues for evaluation infrastructure.

PiperOrigin-RevId: 207820650
parent ade3c79d
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