Commit 9eba75e5 authored by Pavithra Vijay's avatar Pavithra Vijay Committed by TensorFlower Gardener
Browse files

Refactored some of the metrics code in compile function for better readability. 

- Logic change: Moved getting metric name and function out of the training/eval loops in eager mode
- Moved setting metric attributes on the model out the function which calls metric functions.

PiperOrigin-RevId: 213060143
parent 98342d8b
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