Commit 157f8fdb authored by Pavithra Vijay's avatar Pavithra Vijay Committed by TensorFlower Gardener
Browse files

Move convert_to_tensor logic from the `LossFunctionWrapper` class to the... 

Move convert_to_tensor logic from the `LossFunctionWrapper` class to the individual loss functions to support use cases where custom loss functions have non-tensor params.

PiperOrigin-RevId: 231484908
parent 68955444
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